Skip to content

Commit 9e2d940

Browse files
committed
reorganise imports
1 parent d5ac25e commit 9e2d940

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

awswrangler/s3/_write_text.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,7 @@
1818
from awswrangler.s3._write import _COMPRESSION_2_EXT, _apply_dtype, _sanitize, _validate_args
1919
from awswrangler.s3._write_dataset import _to_dataset
2020

21-
if config.memory_format == MemoryFormat.MODIN.value and config.execution_engine == ExecutionEngine.RAY.value:
22-
from modin.pandas import DataFrame as ModinDataFrame
21+
if config.execution_engine == ExecutionEngine.RAY.value:
2322
from ray.data import from_modin, from_pandas
2423
from ray.data.datasource.file_based_datasource import DefaultBlockWritePathProvider
2524

@@ -30,6 +29,9 @@
3029
UserProvidedKeyBlockWritePathProvider,
3130
)
3231

32+
if config.memory_format == MemoryFormat.MODIN.value:
33+
from modin.pandas import DataFrame as ModinDataFrame
34+
3335
_logger: logging.Logger = logging.getLogger(__name__)
3436

3537

0 commit comments

Comments
 (0)