Skip to content

Commit 5df52f3

Browse files
committed
fix test to match
1 parent 624eec7 commit 5df52f3

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/modules/scoring/test_catboost.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ def test_catboost_prediction(dataset):
5555
data_handler = DataHandler(dataset)
5656

5757
scorer = CatBoostScorer(
58-
classification_model_config="prajjwal1/bert-tiny",
58+
embedder_config="prajjwal1/bert-tiny",
5959
iterations=50,
6060
learning_rate=0.05,
6161
depth=6,
@@ -95,7 +95,7 @@ def test_catboost_prediction_multilabel(dataset):
9595
data_handler = DataHandler(dataset.to_multilabel())
9696

9797
scorer = CatBoostScorer(
98-
classification_model_config="prajjwal1/bert-tiny",
98+
embedder_config="prajjwal1/bert-tiny",
9999
iterations=50,
100100
learning_rate=0.05,
101101
depth=6,

0 commit comments

Comments
 (0)