Skip to content

Commit

Permalink
Do not rewrite tags param on create_run()
Browse files Browse the repository at this point in the history
Signed-off-by: Matheus Moreno <matheus.moreno@hurb.com>
  • Loading branch information
Matheus Moreno committed Dec 22, 2021
1 parent 3c85023 commit 355fcd4
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions mlflow/tracking/_tracking_service/client.py
Expand Up @@ -94,18 +94,18 @@ def create_run(self, experiment_id, start_time=None, tags=None):
:return: :py:class:`mlflow.entities.Run` that was created.
"""

tags = tags if tags else {}
run_tags = tags if tags else {}

# Extract user from tags
# This logic is temporary; the user_id attribute of runs is deprecated and will be removed
# in a later release.
user_id = tags.get(MLFLOW_USER, "unknown")
user_id = run_tags.get(MLFLOW_USER, "unknown")

return self.store.create_run(
experiment_id=experiment_id,
user_id=user_id,
start_time=start_time or int(time.time() * 1000),
tags=[RunTag(key, value) for (key, value) in tags.items()],
tags=[RunTag(key, value) for (key, value) in run_tags.items()],
)

def list_run_infos(
Expand Down

0 comments on commit 355fcd4

Please sign in to comment.