import statements modules

kyegomez · Oct 2, 2023 · 45d9863 · 45d9863
1 parent 12a7267
commit 45d9863
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/zeta/__init__.py b/zeta/__init__.py
@@ -3,7 +3,7 @@
 
 
 #nn
-import zeta.nn
+from zeta import nn
 from zeta.nn.architecture.transformer import FeedForward
 from zeta.nn.modules.layernorm import LayerNorm
 
@@ -16,13 +16,13 @@
 
 
 #training
-import zeta.training
+from zeta import training
 
 #tokenizers
-import zeta.tokenizers
+from zeta import tokenizers
 
 #rl
-import zeta.rl
+from zeta import rl
 
 #optim
 import zeta.optim