From ee83505fcc8b4df3978008dcd9921ebd708b4e94 Mon Sep 17 00:00:00 2001
From: George Hotz <72895+geohot@users.noreply.github.com>
Date: Wed, 17 Jan 2024 11:58:08 -0800
Subject: [PATCH] fix test extra issue (#3159)

---
 test/test_multitensor.py                       | 2 ++
 test/{extra => testextra}/__init__.py          | 0
 test/{extra => testextra}/test_export_model.py | 0
 test/{extra => testextra}/test_lr_scheduler.py | 0
 4 files changed, 2 insertions(+)
 rename test/{extra => testextra}/__init__.py (100%)
 rename test/{extra => testextra}/test_export_model.py (100%)
 rename test/{extra => testextra}/test_lr_scheduler.py (100%)

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