From a2a0b20084697407b3e12c6446e46a4f28b80434 Mon Sep 17 00:00:00 2001
From: bdvllrs <bdvllrs@gmail.com>
Date: Wed, 6 Mar 2024 11:18:26 +0000
Subject: [PATCH] Add GWEncoderLinear to __all__ in __init__.py

---
 shimmer/__init__.py         | 2 ++
 shimmer/modules/__init__.py | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/shimmer/__init__.py b/shimmer/__init__.py
index b2699cd2..124065a2 100644
--- 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
index 698ed87a..bcd0a6af 100644
--- 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",