diff --git a/hamilton/plugins/polars_extensions.py b/hamilton/plugins/polars_extensions.py index c8fd7b8a8..6c46eb664 100644 --- a/hamilton/plugins/polars_extensions.py +++ b/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() diff --git a/hamilton/plugins/polars_post_1_0_0_extension.py b/hamilton/plugins/polars_post_1_0_0_extensions.py similarity index 99% rename from hamilton/plugins/polars_post_1_0_0_extension.py rename to hamilton/plugins/polars_post_1_0_0_extensions.py index da9a5475e..666185001 100644 --- a/hamilton/plugins/polars_post_1_0_0_extension.py +++ b/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 diff --git a/tests/plugins/test_polars_extensions.py b/tests/plugins/test_polars_extensions.py index b550c8a09..e5a151b83 100644 --- a/tests/plugins/test_polars_extensions.py +++ b/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, diff --git a/tests/plugins/test_polars_lazyframe_extensions.py b/tests/plugins/test_polars_lazyframe_extensions.py index c5d4df7c8..8899250c9 100644 --- a/tests/plugins/test_polars_lazyframe_extensions.py +++ b/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,