From d086325b1bf9cbd811ea97028bee38dd614860c7 Mon Sep 17 00:00:00 2001 From: George Hotz Date: Mon, 18 Dec 2023 21:12:42 -0800 Subject: [PATCH] hotfix: failing tests --- test/unit/test_shapetracker_math.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/unit/test_shapetracker_math.py b/test/unit/test_shapetracker_math.py index 40e65a3edc..f4cd94361c 100644 --- a/test/unit/test_shapetracker_math.py +++ b/test/unit/test_shapetracker_math.py @@ -16,7 +16,7 @@ class TestShapeTrackerBasics(unittest.TestCase): @unittest.skip("reshape is broken") def test_reshape_makes_same(self): - a = ShapeTracker.from_shape((2, 5, 5)) + a = ShapeTracker.from_shape((2, 5)) x = a.pad( ((2, 0), (0, 0)) ) x = x.reshape( (2, 2, 5) ) x1 = x.reshape( (4, 5) ) @@ -79,7 +79,7 @@ class TestShapeTrackerInvert(unittest.TestCase): @unittest.skip("reshape is broken") def test_invert_failure(self): - a = ShapeTracker.from_shape((2, 5, 5)) + a = ShapeTracker.from_shape((2, 5)) x = a.pad( ((2, 0), (0, 0)) ) x = x.reshape( (2, 2, 5) ) x = x.reshape( (4, 5) )