Update polars dependency requirements

justin13601 · Dec 10, 2024 · 10d36ac · 10d36ac
1 parent 8902365
commit 10d36ac
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,7 +18,7 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 dependencies = [
-    "polars == 1.5.*",
+    "polars >= 1.0.0, <= 1.17.1",
     "bigtree == 0.18.*",
     "ruamel.yaml == 0.18.*",
     "loguru == 0.7.*",

diff --git a/src/aces/predicates.py b/src/aces/predicates.py
@@ -5,6 +5,7 @@
 import polars as pl
 from loguru import logger
 from omegaconf import DictConfig
+from polars.exceptions import ColumnNotFoundError
 
 from .config import TaskExtractorConfig
 from .types import (
@@ -198,7 +199,7 @@ def direct_load_plain_predicates(
 
     missing_columns = [col for col in columns if col not in data.columns]
     if missing_columns:
-        raise pl.ColumnNotFoundError(missing_columns)
+        raise ColumnNotFoundError(missing_columns)
 
     data = data.select(columns)
     ts_type = data.schema["timestamp"]