[CodeBuild] fix test call

exasol · Jan 25, 2024 · 4c07d2e · 4c07d2e
1 parent 1621c6a
commit 4c07d2e
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/integration_tests/without_db/utils/test_load_local_model.py b/tests/integration_tests/without_db/utils/test_load_local_model.py
@@ -62,8 +62,7 @@ def test_load_local_model():
         model.save_pretrained(model_save_path)
         tokenizer.save_pretrained(model_save_path)
 
-        test_setup.loader.load_models(model_name=test_setup.model_name,
-                                      current_model_key=test_setup.mock_current_model_key,
+        test_setup.loader.load_models(current_model_key=test_setup.mock_current_model_key,
                                       model_path=dir_p / "pretrained" / test_setup.model_name)
 
 
@@ -80,6 +79,5 @@ def test_load_local_model_with_huggingface_model_transfer():
         downloaded_model_path = download_model_with_huggingface_transfer(
             test_setup, mock_bucketfs_location)
 
-        test_setup.loader.load_models(model_name=test_setup.model_name,
-                                      current_model_key=test_setup.mock_current_model_key,
+        test_setup.loader.load_models(current_model_key=test_setup.mock_current_model_key,
                                       model_path=downloaded_model_path)