fix tests

GRAAL-Research · Sep 16, 2024 · 72d21cd · 72d21cd
1 parent b9b5079
commit 72d21cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/embeddings_models/multi/multi.wiki.bpe.vs100000.model b/tests/embeddings_models/multi/multi.wiki.bpe.vs100000.model
diff --git a/tests/test_weights_tools.py b/tests/test_weights_tools.py
@@ -22,7 +22,7 @@ def test_givenAS3Path_whenHandleWeights_upload_thenReturnProperWeights(self, tor
 
         handle_weights_upload(path_to_model_to_upload=s3_path)
 
-        torch_mock.has_calls([call.load()])
+        torch_mock.assert_has_calls([call.load()])
 
     @patch("deepparse.weights_tools.CloudPath")
     @patch("deepparse.weights_tools.torch")
@@ -34,7 +34,7 @@ def test_givenAStringS3Path_whenHandleWeights_upload_thenReturnProperWeights(sel
 
         handle_weights_upload(path_to_model_to_upload=s3_path)
 
-        torch_mock.has_calls([call.load()])
+        torch_mock.assert_has_calls([call.load()])
         cloud_path_mock.assert_called()
 
     @patch("deepparse.weights_tools.CloudPath")
@@ -47,7 +47,7 @@ def test_givenAStringPath_whenHandleWeights_upload_thenReturnProperWeights(self,
 
         handle_weights_upload(path_to_model_to_upload=s3_path)
 
-        torch_mock.has_calls([call.load()])
+        torch_mock.assert_has_calls([call.load()])
 
         cloud_path_mock.assert_not_called()