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,
     )