소스 검색

Fixed missed renamed class.

Kristian Schultz 4 년 전
부모
커밋
aeb0ac23d5
2개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  1. 1 2
      library/analysis.py
  2. 1 0
      library/generators/__init__.py

+ 1 - 2
library/analysis.py

@@ -1,7 +1,6 @@
 from library.exercise import Exercise
 from library.dataset import DataSet, TrainTestData
-from library.GanExamples import StupidToyListGan
-from library.generators import SimpleGan, Repeater, SpheredNoise, ConvGAN
+from library.generators import SimpleGan, Repeater, SpheredNoise, ConvGAN, StupidToyListGan
 
 import pickle
 import numpy as np

+ 1 - 0
library/generators/__init__.py

@@ -4,3 +4,4 @@ from library.generators.convGAN import ConvGAN
 from library.generators.LoRAS_ProWRAS import ProWRAS
 from library.generators.Repeater import Repeater
 from library.generators.SpheredNoise import SpheredNoise
+from library.generators.GanExamples import StupidToyListGan, StupidToyPointGan