Skip to content

Commit 1337bb6

Browse files
committed
[skip ci] - minor is_ray_modin in timestream
1 parent 5243dab commit 1337bb6

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

tests/unit/test_timestream.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,10 @@
44
import pytest
55

66
import awswrangler as wr
7-
from awswrangler._distributed import EngineEnum, MemoryFormatEnum
87

9-
if wr.engine.get() == EngineEnum.RAY and wr.memory_format.get() == MemoryFormatEnum.MODIN:
8+
from .._utils import is_ray_modin
9+
10+
if is_ray_modin:
1011
import modin.pandas as pd
1112
else:
1213
import pandas as pd

0 commit comments

Comments
 (0)