demo: hyperpod backend

Author: andrii-i

jupyter_scheduler/backends.py

@@ -57,3 +57,15 @@ class DescribeBackend(BaseModel):
 
     class Config:
         orm_mode = True
+
+
+class SageMakerHyperPodBackend(BaseBackend):
+    """Demo backend simulating SageMaker HyperPod execution."""
+
+    id = "sagemaker_hyperpod"
+    name = "SageMaker HyperPod"
+    description = "Execute Python scripts on SageMaker HyperPod cluster"
+    scheduler_class = "jupyter_scheduler.scheduler.Scheduler"
+    execution_manager_class = "jupyter_scheduler.python_executor.PythonScriptExecutionManager"
+    file_extensions = ["py"]
+    priority = 50

pyproject.toml

@@ -61,6 +61,7 @@ Homepage = "https://github.com/jupyter-server/jupyter-scheduler"
 [project.entry-points."jupyter_scheduler.backends"]
 jupyter_server_nb = "jupyter_scheduler.backends:JupyterServerNotebookBackend"
 jupyter_server_py = "jupyter_scheduler.backends:JupyterServerPythonBackend"
+sagemaker_hyperpod = "jupyter_scheduler.backends:SageMakerHyperPodBackend"
 
 [tool.check-wheel-contents]
 ignore = ["W002"]