Skip to content

Commit 2ec87eb

Browse files
committed
Update Dagster partition imports in iceberg tests
1 parent 3f184e1 commit 2ec87eb

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

libraries/dagster-iceberg/tests/_utils/conftest.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import pyarrow as pa
44
import pytest
5-
from dagster._core.definitions.time_window_partitions import TimeWindow
5+
from dagster._core.definitions.partitions.definition.time_window import TimeWindow
66
from dagster._core.storage.db_io_manager import TablePartitionDimension, TableSlice
77
from pyiceberg import schema as iceberg_schema
88
from pyiceberg import table as iceberg_table

libraries/dagster-iceberg/tests/_utils/test_io.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import pyarrow as pa
66
import pyarrow.compute as pc
77
import pytest
8-
from dagster._core.definitions.time_window_partitions import TimeWindow
8+
from dagster._core.definitions.partitions.definition.time_window import TimeWindow
99
from dagster._core.storage.db_io_manager import TablePartitionDimension, TableSlice
1010
from pyiceberg import expressions as E
1111
from pyiceberg.catalog import Catalog

libraries/dagster-iceberg/tests/_utils/test_partitions.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from unittest import mock
33

44
import pytest
5-
from dagster._core.definitions.time_window_partitions import TimeWindow
5+
from dagster._core.definitions.partitions.definition.time_window import TimeWindow
66
from dagster._core.storage.db_io_manager import TablePartitionDimension, TableSlice
77
from pyiceberg import expressions as E
88
from pyiceberg import partitioning as iceberg_partitioning

0 commit comments

Comments
 (0)