From 56861852be58dfccbde04f5f16b8a45c01fbad07 Mon Sep 17 00:00:00 2001 From: chenyu Date: Sun, 7 Sep 2025 09:06:39 -0400 Subject: [PATCH] enable IMAGE for test_mnist and test_mnist_backward (#12064) passes now --- test/test_tiny.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_tiny.py b/test/test_tiny.py index b6e77d05c9..bf0f903ff2 100644 --- a/test/test_tiny.py +++ b/test/test_tiny.py @@ -99,7 +99,7 @@ class TestTiny(unittest.TestCase): # *** a model *** # TODO: this is failing because of how swizzling rewrites the ShapeTracker of the final STORE - @unittest.skipIf(IMAGE>0 or (CI and Device.DEFAULT == "DSP"), "failing because of make things that can't be images not images") + @unittest.skipIf(CI and Device.DEFAULT == "DSP", "failing because of make things that can't be images not images") def test_mnist(self): layers = [ nn.Conv2d(1, 32, 5), Tensor.relu, @@ -118,7 +118,7 @@ class TestTiny(unittest.TestCase): self.assertEqual(len(probs[0]), 10) # TODO: this is failing because of how swizzling rewrites the ShapeTracker of the final STORE - @unittest.skipIf(IMAGE>0 or (CI and Device.DEFAULT == "DSP"), "failing because of make things that can't be images not images") + @unittest.skipIf(CI and Device.DEFAULT == "DSP", "failing because of make things that can't be images not images") def test_mnist_backward(self): # NOTE: we don't have the whole model here for speed layers = [