remove use of old error handling

Unstructured-IO · Jan 9, 2025 · cb04277 · cb04277
1 parent 9b33680
commit cb04277
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/unstructured_ingest/v2/processes/connectors/databricks/volumes.py b/unstructured_ingest/v2/processes/connectors/databricks/volumes.py
@@ -7,9 +7,6 @@
 
 from pydantic import BaseModel, Field
 
-from unstructured_ingest.error import (
-    SourceConnectionError,
-)
 from unstructured_ingest.utils.dep_check import requires_dependencies
 from unstructured_ingest.v2.errors import (
     ProviderError,
@@ -161,8 +158,7 @@ def precheck(self) -> None:
         try:
             self.connection_config.get_client()
         except Exception as e:
-            logger.error(f"failed to validate connection: {e}", exc_info=True)
-            raise SourceConnectionError(f"failed to validate connection: {e}")
+            raise self.connection_config.wrap_error(e=e)
 
     def get_download_path(self, file_data: FileData) -> Path:
         return self.download_config.download_dir / Path(file_data.source_identifiers.relative_path)