Add GWEncoderLinear to __all__ in __init__.py

ruflab · Mar 6, 2024 · a2a0b20 · a2a0b20
1 parent 3a3f1e0
commit a2a0b20
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/shimmer/__init__.py b/shimmer/__init__.py
@@ -19,6 +19,7 @@
 from shimmer.modules.gw_module import (
     GWDecoder,
     GWEncoder,
+    GWEncoderLinear,
     GWModule,
     GWModuleBase,
     VariationalGWEncoder,
@@ -65,6 +66,7 @@
     "DomainModule",
     "GWDecoder",
     "GWEncoder",
+    "GWEncoderLinear",
     "VariationalGWEncoder",
     "GWModuleBase",
     "GWModule",

diff --git a/shimmer/modules/__init__.py b/shimmer/modules/__init__.py
@@ -19,6 +19,7 @@
 from shimmer.modules.gw_module import (
     GWDecoder,
     GWEncoder,
+    GWEncoderLinear,
     GWModule,
     GWModuleBase,
     VariationalGWEncoder,
@@ -51,6 +52,7 @@
     "DomainModule",
     "GWDecoder",
     "GWEncoder",
+    "GWEncoderLinear",
     "VariationalGWEncoder",
     "GWModuleBase",
     "GWModule",