update

Signed-off-by: Ann Zhang <ann.zhang@databricks.com>

mlflow/tracking/default_experiment/databricks_notebook_experiment_provider.py

@@ -5,6 +5,7 @@
 from mlflow.utils import databricks_utils
 from mlflow.utils.mlflow_tags import (
     MLFLOW_EXPERIMENT_SOURCE_ID,
+    MLFLOW_EXPERIMENT_SOURCE_TYPE
 )
 
 
@@ -24,6 +25,9 @@ def get_experiment_id(self):
             MLFLOW_EXPERIMENT_SOURCE_ID: source_notebook_id,
         }
 
+        if databricks_utils.is_in_databricks_repo_notebook():
+            tags[MLFLOW_EXPERIMENT_SOURCE_TYPE] = "REPO_NOTEBOOK"
+
         # With the presence of the source id, the following is a get or create in which it will
         # return the corresponding experiment if one exists for the repo notebook.
         # For non-repo notebooks, it will raise an exception and we will use source_notebook_id