Exp (mlflow#5509)

Signed-off-by: dbczumar <corey.zumar@databricks.com>
erensahin · Apr 11, 2022 · b78ba38 · b78ba38
1 parent 6f5d5d3
commit b78ba38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlflow/tracking/fluent.py b/mlflow/tracking/fluent.py
@@ -59,7 +59,7 @@
 _logger = logging.getLogger(__name__)
 
 
-def set_experiment(experiment_name: str = None, experiment_id: str = None) -> None:
+def set_experiment(experiment_name: str = None, experiment_id: str = None) -> Experiment:
     """
     Set the given experiment as the active experiment. The experiment must either be specified by
     name via `experiment_name` or by ID via `experiment_id`. The experiment name and ID cannot