diff --git a/integrations/instructor_embedders/tests/test_instructor_backend.py b/integrations/instructor_embedders/tests/test_instructor_backend.py index e2f8a5cc4..85c1f012a 100644 --- a/integrations/instructor_embedders/tests/test_instructor_backend.py +++ b/integrations/instructor_embedders/tests/test_instructor_backend.py @@ -33,7 +33,7 @@ def test_model_initialization(mock_instructor): model="hkunlp/instructor-base", device="cpu", token=Secret.from_token("fake-api-token") ) mock_instructor.assert_called_once_with( - model="hkunlp/instructor-base", device="cpu", use_auth_token="fake-api-token" + model_name_or_path="hkunlp/instructor-base", device="cpu", use_auth_token="fake-api-token" ) # restore the factory state _InstructorEmbeddingBackendFactory._instances = {}