Isolate protected import for pyright

rstudio · Aug 21, 2024 · c021871 · c021871
1 parent 534a1ea
commit c021871
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/pins/drivers.py b/pins/drivers.py
@@ -123,15 +123,14 @@ def load_data(
         elif meta.type == "rds":
             try:
                 import rdata
-
-                # Equivalent to `rdata.read_rds(f)` but compatible with Python 3.8.
-                # See https://github.com/rstudio/pins-python/pull/265
-                parsed = rdata.parser.parse_file(f.name)
-                return rdata.conversion.convert(parsed)
             except ModuleNotFoundError:
                 raise ModuleNotFoundError(
                     "Install the 'rdata' package to attempt to convert 'rds' files into Python objects."
                 )
+            # Equivalent to `rdata.read_rds(f)` but compatible with Python 3.8.
+            # See https://github.com/rstudio/pins-python/pull/265
+            parsed = rdata.parser.parse_file(f.name)
+            return rdata.conversion.convert(parsed)
 
     raise NotImplementedError(f"No driver for type {meta.type}")