support UC directory as meta dataset

Signed-off-by: chenmoneygithub <[email protected]>

mlflow/data/uc_volume_dataset_source.py

@@ -49,9 +49,16 @@ def _verify_uc_path_is_valid(self, path):
             return
 
         try:
+            # Check if path points to a valid UC file.
             w.files.get_metadata(path)
+            return
         except Exception:
-            raise MlflowException(f"{path} does not exist in Databricks Unified Catalog.")
+            try:
+                # Check if path points to a valid UC directory.
+                w.files.get_directory_metadata(path)
+            except Exception:
+                # Neither file nor directory exists, we throw an exception.
+                raise MlflowException(f"{path} does not exist in Databricks Unified Catalog.")
 
     @staticmethod
     def _get_source_type() -> str: