Fix Dask

Signed-off-by: Igoshev, Iaroslav <[email protected]>
YarShev · Apr 24, 2024 · 28b0805 · 28b0805
1 parent 1f4fde4
commit 28b0805
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modin/core/execution/dask/implementations/pandas_on_dask/io/io.py b/modin/core/execution/dask/implementations/pandas_on_dask/io/io.py
@@ -217,11 +217,12 @@ def from_map(cls, func, iterable, *args, **kwargs):
             QueryCompiler containing data returned by map function.
         """
         func = cls.frame_cls._partition_mgr_cls.preprocess_func(func)
+        client = default_client()
         partitions = np.array(
             [
                 [
                     cls.frame_partition_cls(
-                        deploy_map_func.remote(func, obj, *args, **kwargs)
+                        client.submit(deploy_map_func, func, obj, *args, **kwargs)
                     )
                 ]
                 for obj in iterable