|
24 | 24 |
|
25 | 25 | from . import DataFrame |
26 | 26 | from modin.config import IsExperimental, Engine |
27 | | -from modin.core.execution.dispatching.factories.dispatcher import FactoryDispatcher |
28 | 27 | from ...pandas import _update_engine |
29 | 28 |
|
30 | 29 |
|
@@ -96,9 +95,13 @@ def read_sql( |
96 | 95 | ------- |
97 | 96 | modin.DataFrame |
98 | 97 | """ |
| 98 | + _, _, _, kwargs = inspect.getargvalues(inspect.currentframe()) |
| 99 | + |
99 | 100 | Engine.subscribe(_update_engine) |
| 101 | + from modin.core.execution.dispatching.factories.dispatcher import FactoryDispatcher |
| 102 | + |
100 | 103 | assert IsExperimental.get(), "This only works in experimental mode" |
101 | | - _, _, _, kwargs = inspect.getargvalues(inspect.currentframe()) |
| 104 | + |
102 | 105 | return DataFrame(query_compiler=FactoryDispatcher.read_sql(**kwargs)) |
103 | 106 |
|
104 | 107 |
|
@@ -137,9 +140,13 @@ def read_custom_text( |
137 | 140 | ------- |
138 | 141 | modin.DataFrame |
139 | 142 | """ |
| 143 | + _, _, _, kwargs = inspect.getargvalues(inspect.currentframe()) |
| 144 | + |
140 | 145 | Engine.subscribe(_update_engine) |
| 146 | + from modin.core.execution.dispatching.factories.dispatcher import FactoryDispatcher |
| 147 | + |
141 | 148 | assert IsExperimental.get(), "This only works in experimental mode" |
142 | | - _, _, _, kwargs = inspect.getargvalues(inspect.currentframe()) |
| 149 | + |
143 | 150 | return DataFrame(query_compiler=FactoryDispatcher.read_custom_text(**kwargs)) |
144 | 151 |
|
145 | 152 |
|
@@ -265,6 +272,7 @@ def _read(**kwargs) -> DataFrame: |
265 | 272 | [4652013 rows x 18 columns] |
266 | 273 | """ |
267 | 274 | Engine.subscribe(_update_engine) |
| 275 | + from modin.core.execution.dispatching.factories.dispatcher import FactoryDispatcher |
268 | 276 |
|
269 | 277 | try: |
270 | 278 | pd_obj = FactoryDispatcher.read_csv_glob(**kwargs) |
@@ -322,9 +330,13 @@ def read_pickle_distributed( |
322 | 330 | ----- |
323 | 331 | The number of partitions is equal to the number of input files. |
324 | 332 | """ |
| 333 | + _, _, _, kwargs = inspect.getargvalues(inspect.currentframe()) |
| 334 | + |
325 | 335 | Engine.subscribe(_update_engine) |
| 336 | + from modin.core.execution.dispatching.factories.dispatcher import FactoryDispatcher |
| 337 | + |
326 | 338 | assert IsExperimental.get(), "This only works in experimental mode" |
327 | | - _, _, _, kwargs = inspect.getargvalues(inspect.currentframe()) |
| 339 | + |
328 | 340 | return DataFrame(query_compiler=FactoryDispatcher.read_pickle_distributed(**kwargs)) |
329 | 341 |
|
330 | 342 |
|
@@ -369,6 +381,8 @@ def to_pickle_distributed( |
369 | 381 | """ |
370 | 382 | obj = self |
371 | 383 | Engine.subscribe(_update_engine) |
| 384 | + from modin.core.execution.dispatching.factories.dispatcher import FactoryDispatcher |
| 385 | + |
372 | 386 | if isinstance(self, DataFrame): |
373 | 387 | obj = self._query_compiler |
374 | 388 | FactoryDispatcher.to_pickle_distributed( |
|
0 commit comments