From 0fcfc9420c4e884aa57140a04ed5c4772d036289 Mon Sep 17 00:00:00 2001 From: Hemil Desai Date: Tue, 11 Mar 2025 16:25:32 -0700 Subject: [PATCH] Fix docker scheduler creation Signed-off-by: Hemil Desai --- nemo_run/run/torchx_backend/schedulers/docker.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/nemo_run/run/torchx_backend/schedulers/docker.py b/nemo_run/run/torchx_backend/schedulers/docker.py index 131bb0df..ac3a14b3 100644 --- a/nemo_run/run/torchx_backend/schedulers/docker.py +++ b/nemo_run/run/torchx_backend/schedulers/docker.py @@ -18,7 +18,7 @@ import os from datetime import datetime from pathlib import Path -from typing import Iterable, Optional +from typing import Any, Iterable, Optional from torchx.schedulers.api import ( AppDryRunInfo, @@ -272,9 +272,7 @@ def __del__(self) -> None: ) -def create_scheduler( - session_name: str, -) -> PersistentDockerScheduler: +def create_scheduler(session_name: str, **kwargs: Any) -> PersistentDockerScheduler: return PersistentDockerScheduler( session_name=session_name, )