Skip to content

Commit 3fba04f

Browse files
Exclude "nodes" from worker args
Co-authored-by: greptile-apps[bot] <165735046+greptile-apps[bot]@users.noreply.github.com>
1 parent 457fdfd commit 3fba04f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/cloudai/workloads/ai_dynamo/kubernetes_json_gen_strategy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def _to_dynamo_arg(self, arg_name: str) -> str:
154154
return "--" + arg_name.replace("_", "-")
155155

156156
def _dynamo_args_dict(self, model: WorkerBaseArgs) -> dict:
157-
return model.model_dump(exclude={"num_nodes", "extra_args"}, exclude_none=True)
157+
return model.model_dump(exclude={"num_nodes", "extra_args", "nodes"}, exclude_none=True)
158158

159159
def _args_from_worker_config(self, worker: WorkerBaseArgs) -> list[str]:
160160
args = []

0 commit comments

Comments
 (0)