rm old code

llmfoundry/utils/config_utils.py

@@ -28,15 +28,13 @@
     huggingface_dataset_source,
     uc_volume_dataset_source,
 )
-from mlflow.exceptions import MlflowException
 from omegaconf import MISSING, DictConfig, ListConfig, MissingMandatoryValue
 from omegaconf import OmegaConf as om
 from transformers import PretrainedConfig
 
 from llmfoundry.layers_registry import ffns_with_megablocks
 from llmfoundry.models.utils import init_empty_weights
 from llmfoundry.registry import config_transforms
-from llmfoundry.utils.exceptions import UCNotFoundError
 
 log = logging.getLogger(__name__)
 
@@ -798,18 +796,7 @@ def log_dataset_uri(cfg: dict[str, Any]) -> None:
             if dataset_type == 'delta_table':
                 source = source_class(delta_table_name=path)
             elif dataset_type == 'hf' or dataset_type == 'uc_volume':
-                try:
-                    source = source_class(path=path)
-                except MlflowException as e:
-                    error_str = str(e)
-                    match = re.search(
-                        r'MlflowException:\s+(.*?)\s+does not exist in Databricks Unified Catalog\.',
-                        error_str,
-                    )
-                    if match:
-                        uc_path = match.group(1)
-                        raise UCNotFoundError(uc_path)
-                    raise
+                source = source_class(path=path)
             else:
                 source = source_class(url=path)
         else:

llmfoundry/utils/exceptions.py

@@ -33,7 +33,6 @@
     'StoragePermissionError',
     'UCNotEnabledError',
     'DeltaTableNotFoundError',
-    'UCNotFoundError',
 ]
 
 ALLOWED_RESPONSE_KEYS = {'response', 'completion'}
@@ -586,17 +585,3 @@ def __init__(
             volume_name=volume_name,
             table_name=table_name,
         )
-
-
-class UCNotFoundError(UserError):
-    """Error thrown when the UC passed in training doesn't exist."""
-
-    def __init__(
-        self,
-        path: str,
-    ) -> None:
-        message = f'Your data path {path} does not exist. Please double check your UC path'
-        super().__init__(
-            message=message,
-            path=path,
-        )