diff --git a/test/unit/test_gguf.py b/test/unit/test_gguf.py index e9af9dc68e..10b3855cf8 100644 --- a/test/unit/test_gguf.py +++ b/test/unit/test_gguf.py @@ -81,8 +81,7 @@ class TestGGUF(unittest.TestCase): expected.extend(decode(c, E) for c in codes) tensor = Tensor(np.concatenate(blocks)) out = ggml_data_to_tensor(tensor, len(expected), MXFP4) - # TODO: should this be exact equal? somehow failed on CI - np.testing.assert_allclose(out.numpy(), expected, atol=0.0, rtol=1e-6) + np.testing.assert_equal(out.numpy(), expected) def test_expected_failure_unknown_type(self): with self.assertRaises(ValueError):