fix pylint error

microsoft · Dec 13, 2024 · a207d81 · a207d81
1 parent 067ea77
commit a207d81
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qlib/workflow/expm.py b/qlib/workflow/expm.py
@@ -422,11 +422,11 @@ def delete_exp(self, experiment_id=None, experiment_name=None):
 
     def list_experiments(self):
         # retrieve all the existing experiments
-        mlflow_version = int(mlflow.__version__.split(".")[0])
+        mlflow_version = int(mlflow.__version__.split('.', maxsplit=1)[0])
         if mlflow_version >= 2:
             exps = self.client.search_experiments(view_type=ViewType.ACTIVE_ONLY)
         else:
-            exps = self.client.list_experiments(view_type=ViewType.ACTIVE_ONLY)
+            exps = self.client.list_experiments(view_type=ViewType.ACTIVE_ONLY)  # pylint: disable=E1101
         experiments = dict()
         for exp in exps:
             experiment = MLflowExperiment(exp.experiment_id, exp.name, self.uri)