fix: imports

ydataai · Oct 14, 2020 · 7787def · 7787def
1 parent eb2c1fb
commit 7787def
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/src/ydata_synthetic/synthesizers/regular/cgan/model.py b/src/ydata_synthetic/synthesizers/regular/cgan/model.py
@@ -2,6 +2,8 @@
 from os import path
 import numpy as np
 
+from ydata_synthetic.synthesizers import gan
+
 import tensorflow as tf
 from tensorflow.keras.layers import Input, Dense, Dropout, Flatten, Embedding, multiply
 from tensorflow.keras import Model
@@ -10,9 +12,9 @@
 
 class CGAN():
 
-    def __init__(self, gan_args):
+    def __init__(self, model_parameters):
         [self.batch_size, lr, self.noise_dim,
-         self.data_dim, num_classes, self.classes, layers_dim] = gan_args
+         self.data_dim, num_classes, self.classes, layers_dim] = model_parameters
 
         self.generator = Generator(self.batch_size, num_classes). \
             build_model(input_shape=(self.noise_dim,), dim=layers_dim, data_dim=self.data_dim)

diff --git a/src/ydata_synthetic/synthesizers/regular/vanillagan/model.py b/src/ydata_synthetic/synthesizers/regular/vanillagan/model.py
@@ -2,7 +2,7 @@
 from os import path
 import numpy as np
 
-from src.ydata_synthetic.synthesizers import gan
+from ydata_synthetic.synthesizers import gan
 
 import tensorflow as tf
 from tensorflow.keras.layers import Input, Dense, Dropout

diff --git a/src/ydata_synthetic/synthesizers/regular/wgan/model.py b/src/ydata_synthetic/synthesizers/regular/wgan/model.py
@@ -2,7 +2,7 @@
 from os import path
 import numpy as np
 
-from src.ydata_synthetic.synthesizers import gan
+from ydata_synthetic.synthesizers import gan
 
 import tensorflow as tf
 from tensorflow.keras.layers import Input, Dense, Dropout