Skip to content

Commit a9ef804

Browse files
committed
fix: fix SchemaDefinition import error in polars_pre_1_0_0_extension
1 parent e625d65 commit a9ef804

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

hamilton/plugins/polars_pre_1_0_0_extension.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,11 @@
4444

4545
# for polars 0.18.0 we need to check what to do.
4646
if has_alias and hasattr(pl.type_aliases, "CsvEncoding"):
47-
from polars.type_aliases import CsvEncoding, SchemaDefinition
47+
from polars.type_aliases import CsvEncoding
48+
SchemaDefinition = type
4849
else:
4950
CsvEncoding = type
51+
SchemaDefinition = type
5052
if has_alias and hasattr(pl.type_aliases, "CsvQuoteStyle"):
5153
from polars.type_aliases import CsvQuoteStyle
5254
else:

0 commit comments

Comments
 (0)