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 = [