fixed deprecation warning

ion-elgreco · Oct 29, 2024 · 0e2ee3f · 0e2ee3f
1 parent 2f5da97
commit 0e2ee3f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/polars_hash/polars_hash/__init__.py b/polars_hash/polars_hash/__init__.py
@@ -3,10 +3,16 @@
 import warnings
 from pathlib import Path
 from typing import Iterable, Protocol, cast
+from packaging.version import Version
 
 import polars as pl
 from polars.plugins import register_plugin_function
-from polars.type_aliases import IntoExpr, PolarsDataType
+
+if Version(pl.__version__) >= Version("1.0"):
+    from polars._typing import IntoExpr, PolarsDataType
+
+else:
+    from polars.type_aliases import IntoExpr, PolarsDataType
 
 from polars_hash._internal import __version__ as __version__
 

diff --git a/polars_hash/requirements.txt b/polars_hash/requirements.txt
@@ -3,4 +3,5 @@ polars
 pytest
 mypy
 ruff
-black
+black
+packaging
-Original file line number
+Diff line change
@@ Expand Up / @@ -3,4 +3,5 @@ polars @@
     pytest
     mypy
     ruff
-    black
+    black
+    packaging