FEAT-#6990: Implement lazy execution for the Ray virtual partitions.

modin-project · Apr 12, 2024 · b09a944 · b09a944
1 parent 0755a61
commit b09a944
Show file tree

Hide file tree

Showing 9 changed files with 911 additions and 83 deletions.
diff --git a/modin/config/envvars.py b/modin/config/envvars.py
@@ -868,7 +868,7 @@ class LazyExecution(EnvironmentVariable, type=str):
     """
 
     varname = "MODIN_LAZY_EXECUTION"
-    choices = ("Auto", "On", "Off")
+    choices = ("Auto", "On", "Off", "Axis")
     default = "Auto"
 
 

diff --git a/modin/core/execution/ray/common/__init__.py b/modin/core/execution/ray/common/__init__.py
@@ -13,12 +13,18 @@
 
 """Common utilities for Ray execution engine."""
 
-from .engine_wrapper import MaterializationHook, RayWrapper, SignalActor
+from .engine_wrapper import (
+    MaterializationHook,
+    RayObjectRefTypes,
+    RayWrapper,
+    SignalActor,
+)
 from .utils import initialize_ray
 
 __all__ = [
     "initialize_ray",
     "RayWrapper",
     "MaterializationHook",
     "SignalActor",
+    "RayObjectRefTypes",
 ]