Skip to content
This repository was archived by the owner on Apr 15, 2022. It is now read-only.

Commit 5ae291b

Browse files
Epsteinedriggers
authored andcommitted
DBAAS-3012: clean up tags in start_run (#31)
1 parent c422a84 commit 5ae291b

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

splicemachine/ml/management.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -284,7 +284,7 @@ def set_active_run(self, run_id):
284284
"""
285285
self.active_run = self.get_run(run_id)
286286

287-
def start_run(self, tags=None, run_name=None, experiment_id=None, nested=False):
287+
def start_run(self, tags={}, run_name=None, experiment_id=None, nested=False):
288288
"""
289289
Create a new run in the active experiment and set it to be active
290290
:param tags: a dictionary containing metadata about the current run.
@@ -311,9 +311,6 @@ def start_run(self, tags=None, run_name=None, experiment_id=None, nested=False):
311311
raise MlflowException(
312312
"There are no experiements available yet. Please create an experiment before starting a run")
313313

314-
if not tags:
315-
tags = {}
316-
317314
tags['mlflow.user'] = _get_user()
318315

319316
self.active_run = super(MLManager, self).create_run(new_run_exp_id, tags=tags)

0 commit comments

Comments
 (0)