Refined Kubernetes schedular implementation

Author: senseipri

areal/infra/__init__.py

@@ -16,7 +16,7 @@
     RemoteInfBackendProtocol,
     RemoteInfEngine,
 )
-from .scheduler import LocalScheduler, RayScheduler, SlurmScheduler
+from .scheduler import KubernetesScheduler, LocalScheduler, RayScheduler, SlurmScheduler
 from .staleness_manager import StalenessManager
 from .workflow_executor import (
     WorkflowExecutor,
@@ -38,6 +38,7 @@
     "LocalScheduler",
     "RayScheduler",
     "SlurmScheduler",
+    "KubernetesScheduler",
     "LocalLauncher",
     "RayLauncher",
     "SlurmLauncher",

areal/infra/scheduler/__init__.py

@@ -1,5 +1,6 @@
 # SPDX-License-Identifier: Apache-2.0
 
+from .kubernetes import KubernetesScheduler
 from .local import LocalScheduler
 from .ray import RayScheduler
 from .slurm import SlurmScheduler
@@ -8,4 +9,5 @@
     "LocalScheduler",
     "SlurmScheduler",
     "RayScheduler",
+    "KubernetesScheduler",
 ]