Do not use modelId + remove some self.assert (#2405)

* Do not use modelId + remove some self.assert * fix failed test
huggingface · Jul 22, 2024 · 78542a4 · 78542a4
1 parent ac400fe
commit 78542a4
Show file tree

Hide file tree

Showing 4 changed files with 84 additions and 91 deletions.
diff --git a/tests/test_fastai_integration.py b/tests/test_fastai_integration.py
@@ -86,8 +86,8 @@ def test_push_to_hub_and_from_pretrained_fastai(self):
 
         push_to_hub_fastai(learner=dummy_model, repo_id=repo_id, token=TOKEN, config=dummy_config)
         model_info = api.model_info(repo_id)
-        self.assertEqual(model_info.modelId, repo_id)
+        assert model_info.id == repo_id
         loaded_model = from_pretrained_fastai(repo_id)
-        self.assertEqual(dummy_model.show_training_loop(), loaded_model.show_training_loop())
+        assert dummy_model.show_training_loop() == loaded_model.show_training_loop()
 
         api.delete_repo(repo_id=repo_id)