Эх сурвалжийг харах

Fixed missed renamed class.

Kristian Schultz 4 жил өмнө
parent
commit
aeb0ac23d5

+ 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