rename to architectures

databricks · Aug 12, 2024 · 4e1067b · 4e1067b
1 parent 3c59e1c
commit 4e1067b
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/tests/layers/testing.py → tests/layers/architectures.py b/tests/layers/testing.py → tests/layers/architectures.py
diff --git a/tests/layers/dmoe_test.py b/tests/layers/dmoe_test.py
@@ -10,7 +10,7 @@
 from megablocks.layers.arguments import Arguments
 from megablocks.layers.dmoe import dMoE
 from megablocks.layers.moe import MoE, batched_load_balancing_loss, clear_load_balancing_loss
-from tests.layers.testing import FFN
+from tests.layers.architectures import FFN
 
 # min size: (1, 2, 128, 2, 1)
 _FORWARD_TESTS_DEFAULT = (

diff --git a/tests/layers/glu_test.py b/tests/layers/glu_test.py
@@ -9,7 +9,7 @@
 
 from megablocks.layers import dmlp_registry
 from megablocks.layers.arguments import Arguments
-from tests.layers.testing import GLU
+from tests.layers.architectures import GLU
 
 _DENSE_TESTS = (
     (16, 1024, 512),

diff --git a/tests/layers/moe_test.py b/tests/layers/moe_test.py
@@ -8,7 +8,7 @@
 
 from megablocks.layers.arguments import Arguments
 from megablocks.layers.moe import MoE, batched_load_balancing_loss, clear_load_balancing_loss
-from tests.layers.testing import FFN
+from tests.layers.architectures import FFN
 
 _FORWARD_TESTS = (
     (16, 1024, 512, 1, 1),