error earlier

Author: isabelizimm

pins/constructors.py

@@ -10,6 +10,7 @@
 from .boards import BaseBoard, BoardManual, BoardRsConnect, board_deparse
 from .cache import PinsAccessTimeCache, PinsCache, PinsRscCacheMapper, prefix_cache
 from .config import get_cache_dir, get_data_dir
+from .errors import PinsError
 
 # Kept here for backward-compatibility reasons
 # Note that this is not a constructor, but a function to represent them.
@@ -625,5 +626,10 @@ def board_databricks(path, versioned=True, cache=DEFAULT, allow_pickle_read=None
     0  1  a  3
     1  2  b  4
     """
-
+    try:
+        import databricks.sdk  # noqa: F401
+    except ModuleNotFoundError:
+        raise PinsError(
+            "Install the `databricks-sdk` package for Databricks board support."
+        )
     return board("dbc", path, versioned, cache, allow_pickle_read)

pins/databricks/fs.py

@@ -53,12 +53,7 @@ def rm(self, path, recursive=True, maxdepth=None) -> None:
 
     @staticmethod
     def _databricks_put(lpath, rpath):
-        try:
-            from databricks.sdk import WorkspaceClient
-        except ModuleNotFoundError:
-            raise PinsError(
-                "Install the `databricks-sdk` package for Databricks board support."
-            )
+        from databricks.sdk import WorkspaceClient
 
         w = WorkspaceClient()
         path = Path(lpath).absolute()
@@ -80,12 +75,7 @@ def _upload_files(path):
         _upload_files(path)
 
     def _databricks_get(self, board, rpath, lpath, recursive=False, **kwargs):
-        try:
-            from databricks.sdk import WorkspaceClient
-        except ModuleNotFoundError:
-            raise PinsError(
-                "Install the `databricks-sdk` package for Databricks board support."
-            )
+        from databricks.sdk import WorkspaceClient
 
         w = WorkspaceClient()
         file_type = self._databricks_is_type(rpath)
@@ -110,12 +100,7 @@ def _get_files(path, recursive, **kwargs):
         _get_files(rpath, recursive, **kwargs)
 
     def _databricks_open(self, path):
-        try:
-            from databricks.sdk import WorkspaceClient
-        except ModuleNotFoundError:
-            raise PinsError(
-                "Install the `databricks-sdk` package for Databricks board support."
-            )
+        from databricks.sdk import WorkspaceClient
 
         if not self._databricks_exists(path):
             raise PinsError(f"File or directory does not exist at path: {path}")
@@ -134,13 +119,8 @@ def _databricks_exists(self, path: str):
 
     @staticmethod
     def _databricks_is_type(path: str):
-        try:
-            from databricks.sdk import WorkspaceClient
-            from databricks.sdk.errors import NotFound
-        except ModuleNotFoundError:
-            raise PinsError(
-                "Install the `databricks-sdk` package for Databricks board support."
-            )
+        from databricks.sdk import WorkspaceClient
+        from databricks.sdk.errors import NotFound
 
         w = WorkspaceClient()
         try:
@@ -156,12 +136,7 @@ def _databricks_is_type(path: str):
             return "file"
 
     def _databricks_ls(self, path, detail):
-        try:
-            from databricks.sdk import WorkspaceClient
-        except ModuleNotFoundError:
-            raise PinsError(
-                "Install the `databricks-sdk` package for Databricks board support."
-            )
+        from databricks.sdk import WorkspaceClient
 
         if not self._databricks_exists(path):
             raise PinsError(f"File or directory does not exist at path: {path}")
@@ -190,12 +165,7 @@ def _databricks_ls(self, path, detail):
         return items
 
     def _databricks_rm_dir(self, path):
-        try:
-            from databricks.sdk import WorkspaceClient
-        except ModuleNotFoundError:
-            raise PinsError(
-                "Install the `databricks-sdk` package for Databricks board support."
-            )
+        from databricks.sdk import WorkspaceClient
 
         w = WorkspaceClient()
         raw_contents = w.files.list_directory_contents(path)
@@ -211,12 +181,7 @@ def _databricks_rm_dir(self, path):
 
     @staticmethod
     def _databricks_mkdir(path):
-        try:
-            from databricks.sdk import WorkspaceClient
-        except ModuleNotFoundError:
-            raise PinsError(
-                "Install the `databricks-sdk` package for Databricks board support."
-            )
+        from databricks.sdk import WorkspaceClient
 
         w = WorkspaceClient()
         w.files.create_directory(path)