fixup

narwhals-dev · Aug 13, 2024 · b937b5f · b937b5f
1 parent ec970c6
commit b937b5f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/narwhals/dataframe.py b/narwhals/dataframe.py
@@ -263,7 +263,7 @@ def __arrow_c_stream__(self, requested_schema: object | None = None) -> object:
         if hasattr(native_frame, "__arrow_c_stream__"):
             return native_frame.__arrow_c_stream__(requested_schema=requested_schema)
         try:
-            import pyarrow as pa
+            import pyarrow as pa  # ignore-banned-import
         except ModuleNotFoundError as exc:  # pragma: no cover
             msg = f"PyArrow>=14.0.0 is required for `__arrow_c_stream__` for object of type {type(native_frame)}"
             raise ModuleNotFoundError(msg) from exc

diff --git a/narwhals/series.py b/narwhals/series.py
@@ -74,7 +74,7 @@ def __arrow_c_stream__(self, requested_schema: object | None = None) -> object:
         if hasattr(native_series, "__arrow_c_stream__"):
             return native_series.__arrow_c_stream__(requested_schema=requested_schema)
         try:
-            import pyarrow as pa
+            import pyarrow as pa  # ignore-banned-import
         except ModuleNotFoundError as exc:  # pragma: no cover
             msg = f"PyArrow>=14.0.0 is required for `__arrow_c_stream__` for object of type {type(native_series)}"
             raise ModuleNotFoundError(msg) from exc