add a missing unittest.main() [pr] (#8274)

This commit is contained in:
chenyu 2024-12-16 14:28:10 -05:00 committed by GitHub
commit 2bb298f38d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 1 deletions

View file

@ -102,3 +102,6 @@ class TestGGUF(unittest.TestCase):
if (get_fn := gguf_val_getters[v_type]) is not None: self.assertEqual(get_fn(gguf_ctx, gguf_key_id), v)
ggml.gguf_free(gguf_ctx)
if __name__ == '__main__':
unittest.main()

View file

@ -182,4 +182,3 @@ class TestShapeTrackerInvert(unittest.TestCase):
if __name__ == '__main__':
unittest.main()