mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

hotfix: failing tests

Browse source
This commit is contained in:
George Hotz 2023-12-18 21:12:42 -08:00
commit d086325b1b
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
test/unit/test_shapetracker_math.py
View file

@ -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) )