Skip to content

Commit 5a641a6

Browse files
committed
minor bug fix
1 parent f092999 commit 5a641a6

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/modules/scoring/test_description.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ def test_description_scorer(dataset, expected_prediction, multilabel):
1717
db_dir, dump_dir, logs_dir = setup_environment()
1818
data_handler = DataHandler(dataset, force_multilabel=multilabel)
1919

20-
scorer = DescriptionScorer(embedder_name="sergeyzh/rubert-tiny-turbo", db_dir=db_dir, temperature=0.3, device="cpu")
20+
scorer = DescriptionScorer(embedder_name="sergeyzh/rubert-tiny-turbo", temperature=0.3, device="cpu")
2121

2222
scorer.fit(data_handler.utterances_train, data_handler.labels_train, data_handler.label_description)
2323
assert scorer.description_vectors.shape[0] == len(data_handler.label_description)

0 commit comments

Comments
 (0)