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

Author: AndreyPavlenko

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"
 
 

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",
 ]