slight adjustment

Author: rithwik-db

compose_rl/utils/mlflow_utils.py

@@ -27,17 +27,14 @@ def get_valid_mlflow_experiment_name(config: Any) -> str:
 
 
 def get_mlflow_relative_path_for_save_folder(save_folder: str) -> str:
-    """Returns the relative path for the given save folder
-
-    Relative in mlflow need to be of the format: `artifacts/{relative_path}`
-    """
-    return os.path.join('artifacts', save_folder.lstrip('/'))
+    """Returns the relative path for the given save folder"""
+    return save_folder.lstrip('/')
 
 
 def get_mlflow_absolute_path_for_save_folder(save_folder: str) -> str:
     """Returns the mlflow artifact path for the given save folder"""
     mlflow_prefix = 'dbfs:/databricks/mlflow-tracking/{mlflow_experiment_id}/{mlflow_run_id}'
-    mlflow_artifact_path = os.path.join(mlflow_prefix, get_mlflow_relative_path_for_save_folder(save_folder))
+    mlflow_artifact_path = os.path.join(mlflow_prefix, 'artifacts', get_mlflow_relative_path_for_save_folder(save_folder))
     return mlflow_artifact_path