[FEATS] [Structs]

kyegomez · Jan 21, 2024 · 92976a2 · 92976a2
1 parent 530880b
commit 92976a2
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/swarms_torch/structs/__init__.py b/swarms_torch/structs/__init__.py
@@ -1,5 +1,11 @@
 from swarms_torch.structs.parallel_wrapper import ParallelSwarm
 from swarms_torch.structs.moe import SwitchGate, SwitchMoE
-from swarms_torch.structs.simple_moe import SimpleMoE
+from swarms_torch.structs.simple_moe import GatingMechanism, SimpleMoE
 
-__all__ = ["ParallelSwarm", "SwitchGate", "SwitchMoE", "SimpleMoE"]
+__all__ = [
+    "ParallelSwarm",
+    "SwitchGate",
+    "SwitchMoE",
+    "GatingMechanism",
+    "SimpleMoE",
+]