Skip to content

Commit 5781974

Browse files
committed
fix: scoring test
1 parent 56485da commit 5781974

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/nodes/test_scoring.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def test_scoring_multiclass(embedding_optimizer_multiclass):
4545
},
4646
{
4747
"module_name": "description",
48-
"temperature": [1.0, 0.5, 0.1, 0.05],
48+
"temperature": [1.0],
4949
"embedder_config": ["sergeyzh/rubert-tiny-turbo"],
5050
"encoder_type": ["bi", "cross"],
5151
"cross_encoder_config": ["cross-encoder/ms-marco-MiniLM-L-6-v2"],
@@ -108,7 +108,7 @@ def test_scoring_multilabel(embedding_optimizer_multilabel):
108108
},
109109
{
110110
"module_name": "description",
111-
"temperature": [1.0, 0.5, 0.1, 0.05],
111+
"temperature": [1.0],
112112
"embedder_config": ["sergeyzh/rubert-tiny-turbo"],
113113
"encoder_type": ["bi", "cross"],
114114
"cross_encoder_config": ["cross-encoder/ms-marco-MiniLM-L-6-v2"],

0 commit comments

Comments
 (0)