use run_name in FileStore when provided (mlflow#358)

ljzzju · Aug 24, 2018 · 4160710 · 4160710
1 parent 2037d2c
commit 4160710
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mlflow/store/file_store.py b/mlflow/store/file_store.py
@@ -211,7 +211,8 @@ def create_run(self, experiment_id, user_id, run_name, source_type,
         run_uuid = uuid.uuid4().hex
         artifact_uri = self._get_artifact_dir(experiment_id, run_uuid)
         num_runs = len(self._list_run_uuids(experiment_id))
-        run_info = RunInfo(run_uuid=run_uuid, experiment_id=experiment_id, name="Run %s" % num_runs,
+        run_info = RunInfo(run_uuid=run_uuid, experiment_id=experiment_id,
+                           name=run_name or "Run %s" % num_runs,
                            artifact_uri=artifact_uri, source_type=source_type,
                            source_name=source_name,
                            entry_point_name=entry_point_name, user_id=user_id,