fix test_itertuples

snowflakedb · Sep 19, 2024 · a36f07e · a36f07e
1 parent 72ce905
commit a36f07e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/integ/modin/frame/test_itertuples.py b/tests/integ/modin/frame/test_itertuples.py
@@ -8,7 +8,9 @@
 import pytest
 
 import snowflake.snowpark.modin.plugin  # noqa: F401
-from snowflake.snowpark.modin.pandas.snow_partition_iterator import PARTITION_SIZE
+from snowflake.snowpark.modin.plugin.extensions.snow_partition_iterator import (
+    PARTITION_SIZE,
+)
 from tests.integ.modin.sql_counter import SqlCounter, sql_count_checker
 from tests.integ.modin.utils import eval_snowpark_pandas_result
 from tests.utils import running_on_public_ci