fix: updated imports for polars extension 1.0­.0

hamilton/plugins/polars_extensions.py

@@ -26,7 +26,7 @@
         f"Current version: {pl_version}, minimum required version: 1.0.0."
     )
 else:
-    from hamilton.plugins.polars_post_1_0_0_extension import register_data_loaders
+    from hamilton.plugins.polars_post_1_0_0_extensions import register_data_loaders
 
 register_data_loaders()
 

hamilton/plugins/polars_post_1_0_0_extension.py → hamilton/plugins/polars_post_1_0_0_extensions.py

@@ -593,8 +593,8 @@ class PolarsSpreadsheetWriter(DataSaver):
     # importing here because this is where it's used. Can move later.
     # but yeah the polars type aliases weren't resolving well in python 3.9
     # so stripped/reduced them appropriately.
+    from polars._typing import ColumnTotalsDefinition, RowTotalsDefinition
     from polars.datatypes import DataType, DataTypeClass
-    from polars.type_aliases import ColumnTotalsDefinition, RowTotalsDefinition
 
     workbook: Union[Workbook, BytesIO, Path, str]
     worksheet: Union[str, None] = None

tests/plugins/test_polars_extensions.py

@@ -9,7 +9,7 @@
 import polars as pl  # isort: skip
 import pytest  # isort: skip
 
-from hamilton.plugins.polars_post_1_0_0_extension import (  # isort: skip
+from hamilton.plugins.polars_post_1_0_0_extensions import (  # isort: skip
     PolarsAvroReader,
     PolarsAvroWriter,
     PolarsCSVReader,

tests/plugins/test_polars_lazyframe_extensions.py

@@ -11,7 +11,7 @@
     PolarsScanFeatherReader,
     PolarsScanParquetReader,
 )
-from hamilton.plugins.polars_post_1_0_0_extension import (
+from hamilton.plugins.polars_post_1_0_0_extensions import (
     PolarsAvroReader,
     PolarsAvroWriter,
     PolarsCSVWriter,