浏览代码

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