Optional import fixes

tconbeer · Jul 9, 2024 · 9423fe3 · 9423fe3
1 parent e68887e
commit 9423fe3
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -17,6 +17,7 @@ repos:
           - pytest
           - pyarrow-stubs
           - pandas-stubs
+          - polars
         exclude: "tests/snapshot_tests/"
         args:
           - "--disallow-untyped-calls"

diff --git a/pyproject.toml b/pyproject.toml
@@ -42,6 +42,7 @@ pandas-stubs = { version="^2.1.1", python=">=3.9,<3.13" }
 pytest = "^7.3.1"
 pytest-asyncio = ">=0.21,<0.24"
 pytest-textual-snapshot = ">=0.4.0"
+polars = ">=0.20.0"
 
 
 [tool.ruff]

diff --git a/src/textual_fastdatatable/__init__.py b/src/textual_fastdatatable/__init__.py
@@ -1,6 +1,5 @@
 from textual_fastdatatable.backend import (
     ArrowBackend,
-    PolarsBackend,
     DataTableBackend,
     create_backend,
 )
@@ -9,7 +8,6 @@
 __all__ = [
     "DataTable",
     "ArrowBackend",
-    "PolarsBackend",
     "DataTableBackend",
     "create_backend",
 ]
diff --git a/src/textual_fastdatatable/backend.py b/src/textual_fastdatatable/backend.py
@@ -53,7 +53,7 @@ def create_backend(
         return ArrowBackend(data, max_rows=max_rows)
     if isinstance(data, pa.RecordBatch):
         return ArrowBackend.from_batches(data, max_rows=max_rows)
-    if isinstance(data, pl.DataFrame) and _HAS_POLARS:
+    if _HAS_POLARS and isinstance(data, pl.DataFrame):
         return PolarsBackend.from_dataframe(data, max_rows=max_rows)
 
     if isinstance(data, Path) or isinstance(data, str):