diff --git a/pytorch_lightning/logging/mlflow.py b/pytorch_lightning/logging/mlflow.py index 50f4843e..bf04917c 100644 --- a/pytorch_lightning/logging/mlflow.py +++ b/pytorch_lightning/logging/mlflow.py @@ -47,6 +47,7 @@ class MLFlowLogger(LightningLoggerBase): experiment_name (str): The name of the experiment tracking_uri (str): where this should track tags (dict): todo this param + """ super().__init__() self._mlflow_client = mlflow.tracking.MlflowClient(tracking_uri)