Skip to content

Commit

Permalink
fix
Browse files Browse the repository at this point in the history
  • Loading branch information
anakin87 committed Feb 14, 2024
1 parent ad15e25 commit 2f66cdf
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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 = {}
Expand Down

0 comments on commit 2f66cdf

Please sign in to comment.