FIX: bugs in test_command_line

RUCAIBox · Aug 31, 2021 · 9f80803 · 9f80803
1 parent dd3d7b6
commit 9f80803
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/python-package.yml b/.github/workflows/python-package.yml
@@ -55,5 +55,5 @@ jobs:
       run: |
         python -m pytest -v tests/config/test_config.py
         export PYTHONPATH=.
-        python tests/config/test_command_line.py --use_gpu=False --valid_metric=Recall@10 --split_ratio=[0.7,0.2,0.1] --metrics=['Recall'] --topk=[10] --epochs=200 --eval_setting='LO_RS' --learning_rate=0.3
+        python tests/config/test_command_line.py --use_gpu=False --valid_metric=Recall@10 --split_ratio=[0.7,0.2,0.1] --metrics='["Recall"]' --topk=[10] --epochs=200 --eval_setting='LO_RS' --learning_rate=0.3
 
diff --git a/docs/source/developer_guide/customize_metrics.rst b/docs/source/developer_guide/customize_metrics.rst
@@ -19,7 +19,7 @@ Set properties of the metric
 -----------------------------
 After that, we set the properties of metrics:
 
-Set ``metric_head``
+Set ``metric_need``
 ###################
 It is a list that contains one or multiple string that corresponding to needed input of metrics.
 For now, we support 9 inputs for metrics including both recommendation results and information of

diff --git a/tests/config/run.sh b/tests/config/run.sh
@@ -4,5 +4,5 @@
 python -m unittest
 
 python test_command_line.py --use_gpu=False --valid_metric=Recall@10 --split_ratio=[0.7,0.2,0.1] \
---metrics=['Recall'] --topk=[10] \
+--metrics='["Recall"]' --topk=[10] \
 --epochs=200 --eval_setting='LO_RS' --learning_rate=0.3