feat(synth): Changes to init files.

ydataai · Jan 25, 2021 · e6cba5f · e6cba5f
1 parent 64a6f6f
commit e6cba5f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 13 deletions.
diff --git a/src/ydata_synthetic/synthesizers/__init__.py b/src/ydata_synthetic/synthesizers/__init__.py
@@ -1,13 +0,0 @@
-from ydata_synthetic.synthesizers.regular.cgan.model import CGAN
-from ydata_synthetic.synthesizers.regular.wgan.model import WGAN
-from ydata_synthetic.synthesizers.regular.vanillagan.model import VanilllaGAN
-from ydata_synthetic.synthesizers.regular.wgangp.model import WGAN_GP
-from ydata_synthetic.synthesizers.time_series.timegan.model import TimeGAN
-
-__all__ = [
-    "VanilllaGAN",
-    "CGAN",
-    "WGAN",
-    "WGAN_GP",
-    "TimeGAN"
-]

diff --git a/src/ydata_synthetic/synthesizers/regular/__init__.py b/src/ydata_synthetic/synthesizers/regular/__init__.py
@@ -0,0 +1,11 @@
+from ydata_synthetic.synthesizers.regular.cgan.model import CGAN
+from ydata_synthetic.synthesizers.regular.wgan.model import WGAN
+from ydata_synthetic.synthesizers.regular.vanillagan.model import VanilllaGAN
+from ydata_synthetic.synthesizers.regular.wgangp.model import WGAN_GP
+
+__all__ = [
+    "VanilllaGAN",
+    "CGAN",
+    "WGAN",
+    "WGAN_GP"
+]