Skip to content

Commit ebfdd3e

Browse files
committed
fix: fixed unit test
1 parent f62bba9 commit ebfdd3e

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

tests/callback/test_callback.py

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,7 @@ def test_pipeline_callbacks(dataset):
4848
{
4949
"node_type": "embedding",
5050
"target_metric": "retrieval_hit_rate",
51+
"metrics": ["retrieval_map", "retrieval_mrr", "retrieval_ndcg", "retrieval_precision"],
5152
"search_space": [
5253
{
5354
"module_name": "retrieval",
@@ -59,6 +60,13 @@ def test_pipeline_callbacks(dataset):
5960
{
6061
"node_type": "scoring",
6162
"target_metric": "scoring_roc_auc",
63+
"metrics": [
64+
"scoring_accuracy",
65+
"scoring_f1",
66+
"scoring_log_likelihood",
67+
"scoring_precision",
68+
"scoring_recall",
69+
],
6270
"search_space": [
6371
{"module_name": "knn", "k": [1], "weights": ["uniform", "distance"]},
6472
{"module_name": "linear"},
@@ -67,6 +75,13 @@ def test_pipeline_callbacks(dataset):
6775
{
6876
"node_type": "decision",
6977
"target_metric": "decision_accuracy",
78+
"metrics": [
79+
"decision_accuracy",
80+
"decision_f1",
81+
"decision_precision",
82+
"decision_recall",
83+
"decision_roc_auc",
84+
],
7085
"search_space": [{"module_name": "threshold", "thresh": [0.5]}, {"module_name": "argmax"}],
7186
},
7287
]

0 commit comments

Comments
 (0)