[TESTS][Fixed]

tests/structs/test_encoderdecoder.py

@@ -2,7 +2,7 @@
 import argparse
 import pytest
 
-from zeta.nn import EncoderDecoder, Encoder, Decoder
+from zeta.structs import EncoderDecoder, Encoder, Decoder
 
 
 @pytest.fixture

tests/structs/test_hierarchicalblock.py

@@ -1,6 +1,6 @@
 import pytest
 import torch
-from zeta.nn import HierarchicalBlock
+from zeta.structs import HierarchicalBlock
 
 
 def test_HierarchicalBlock_init():

tests/structs/test_localtransformer.py

@@ -1,7 +1,7 @@
 from torch import nn
 import pytest
 import torch
-from zeta.nn import LocalTransformer
+from zeta.structs import LocalTransformer
 from torch.autograd import gradcheck
 from zeta.nn.modules.dynamic_module import DynamicPositionBias
 

tests/structs/test_paralleltransformerblock.py

@@ -1,6 +1,6 @@
 import torch
 import pytest
-from zeta.nn import ParallelTransformerBlock
+from zeta.structs import ParallelTransformerBlock
 from torch.autograd import gradcheck
 
 

tests/structs/test_simpletransformer.py

@@ -1,7 +1,7 @@
 import pytest
 import torch
 import torch.nn as nn
-from zeta.nn import SimpleTransformer
+from zeta.structs import SimpleTransformer
 
 
 def test_valid_init():

tests/structs/test_transformer.py

@@ -1,6 +1,6 @@
 import pytest
 import torch
-from zeta.nn import Transformer, AttentionLayers
+from zeta.structs import Transformer, AttentionLayers
 
 # assuming that you are testing the Transformer class
 

tests/structs/test_vitransformerwrapper.py

@@ -1,6 +1,6 @@
 import pytest
 import torch
-from zeta.nn import ViTransformerWrapper, Encoder
+from zeta.structs import ViTransformerWrapper, Encoder
 from torch.nn import Module