diff --git a/test/test_multitensor.py b/test/test_multitensor.py index 21a4bbbeb1..4e716394bb 100644 --- a/test/test_multitensor.py +++ b/test/test_multitensor.py @@ -24,12 +24,14 @@ class TestMultiTensor(unittest.TestCase): X.to_((d0, d1)) for lb in X.lazydata.lbs: assert lb.shape == (256,) + (X + X).realize() def test_shard(self): X = Tensor.ones(256).contiguous().realize() X.shard_((d0, d1), 0) for lb in X.lazydata.lbs: assert lb.shape == (128,) + (X + X).realize() def test_shard_same_device(self): X = Tensor.ones(256).contiguous().realize() diff --git a/test/extra/__init__.py b/test/testextra/__init__.py similarity index 100% rename from test/extra/__init__.py rename to test/testextra/__init__.py diff --git a/test/extra/test_export_model.py b/test/testextra/test_export_model.py similarity index 100% rename from test/extra/test_export_model.py rename to test/testextra/test_export_model.py diff --git a/test/extra/test_lr_scheduler.py b/test/testextra/test_lr_scheduler.py similarity index 100% rename from test/extra/test_lr_scheduler.py rename to test/testextra/test_lr_scheduler.py