From 28cb827a23120fa7dcbfadb5a4c8119f33d49359 Mon Sep 17 00:00:00 2001 From: chaosyu Date: Thu, 15 Jul 2021 11:46:10 +0800 Subject: [PATCH] fix lint issue --- qlib/workflow/exp.py | 6 ++++-- qlib/workflow/task/collect.py | 2 +- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/qlib/workflow/exp.py b/qlib/workflow/exp.py index 63ab3ce40..fcf6cd8d1 100644 --- a/qlib/workflow/exp.py +++ b/qlib/workflow/exp.py @@ -325,7 +325,7 @@ class MLflowExperiment(Experiment): UNLIMITED = 50000 # FIXME: Mlflow can only list 50000 records at most!!!!!!! - def list_recorders(self, max_results: int = UNLIMITED, status: Union[str, None] = None, filter_string: str=""): + def list_recorders(self, max_results: int = UNLIMITED, status: Union[str, None] = None, filter_string: str = ""): """ Parameters ---------- @@ -337,7 +337,9 @@ class MLflowExperiment(Experiment): filter_string : str mlflow supported filter string like 'params."my_param"="a" and tags."my_tag"="b"', use this will help to reduce too much run number. """ - runs = self._client.search_runs(self.id, run_view_type=ViewType.ACTIVE_ONLY, max_results=max_results, filter_string=filter_string) + runs = self._client.search_runs( + self.id, run_view_type=ViewType.ACTIVE_ONLY, max_results=max_results, filter_string=filter_string + ) recorders = dict() for i in range(len(runs)): recorder = MLflowRecorder(self.id, self._uri, mlflow_run=runs[i]) diff --git a/qlib/workflow/task/collect.py b/qlib/workflow/task/collect.py index c7d82d541..6f92034aa 100644 --- a/qlib/workflow/task/collect.py +++ b/qlib/workflow/task/collect.py @@ -139,7 +139,7 @@ class RecorderCollector(Collector): rec_filter_func=None, artifacts_path={"pred": "pred.pkl"}, artifacts_key=None, - filter_string: str = "" + filter_string: str = "", ): """ Init RecorderCollector.