zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 04:14:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

work with tuple shape (#393)

Browse Source
This commit is contained in:
Awni Hannun
2024-02-01 13:03:47 -08:00
committed by GitHub
parent 0340113e02
commit ec14583c2a
6 changed files with 5 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
whisper/test.py
View File

@@ -303,7 +303,7 @@ class TestWhisper(unittest.TestCase):
def check_segment(seg, expected):
for k, v in expected.items():
if isinstance(v, float):
self.assertAlmostEqual(seg[k], v, places=3)
self.assertAlmostEqual(seg[k], v, places=2)
else:
self.assertEqual(seg[k], v)