Merge pull request #8 from will-rice/wr-fix-imports

Fix top-level imports
will-rice · Oct 11, 2023 · dee4e87 · dee4e87
2 parents 94973fc + 2c23c33
commit dee4e87
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/denoisers/__init__.py b/denoisers/__init__.py
@@ -1,5 +1,5 @@
 """Denoisers for the 1D and 2D cases."""
-from modeling.waveunet.config import WaveUNetConfig
-from modeling.waveunet.model import WaveUNetModel
+from denoisers.modeling.waveunet.config import WaveUNetConfig
+from denoisers.modeling.waveunet.model import WaveUNetModel
 
 __all__ = ["WaveUNetConfig", "WaveUNetModel"]
diff --git a/denoisers/modeling/__init__.py b/denoisers/modeling/__init__.py
@@ -1 +1,5 @@
 """Models."""
+from denoisers.modeling.waveunet.config import WaveUNetConfig
+from denoisers.modeling.waveunet.model import WaveUNetModel
+
+__all__ = ["WaveUNetConfig", "WaveUNetModel"]
diff --git a/denoisers/modeling/waveunet/__init__.py b/denoisers/modeling/waveunet/__init__.py
@@ -1 +1,5 @@
 """1d UNet model."""
+from denoisers.modeling.waveunet.config import WaveUNetConfig
+from denoisers.modeling.waveunet.model import WaveUNetModel
+
+__all__ = ["WaveUNetConfig", "WaveUNetModel"]