From c80cabef015e5ffbb1739c5d5d6e060f56a75c51 Mon Sep 17 00:00:00 2001 From: kerthcet Date: Thu, 27 Nov 2025 00:37:27 +0000 Subject: [PATCH] Support max_iterations Signed-off-by: kerthcet --- src/hive_cli/config.py | 5 +++++ src/hive_cli/platform/k8s.py | 1 + 2 files changed, 6 insertions(+) diff --git a/src/hive_cli/config.py b/src/hive_cli/config.py index 623266d..cd79f64 100644 --- a/src/hive_cli/config.py +++ b/src/hive_cli/config.py @@ -133,6 +133,11 @@ class RuntimeConfig(BaseModel): description="Maximum runtime for the experiment in seconds. \ -1 means no limit.", ) + max_iterations: int = Field( + default=-1, + description="Maximum number of iterations for the experiment. \ + -1 means no limit.", + ) class HiveConfig(BaseModel): diff --git a/src/hive_cli/platform/k8s.py b/src/hive_cli/platform/k8s.py index 933597d..769a47d 100644 --- a/src/hive_cli/platform/k8s.py +++ b/src/hive_cli/platform/k8s.py @@ -271,6 +271,7 @@ def construct_experiment(name: str, namespace: str, config: HiveConfig) -> dict: "runtime": { "numAgents": config.runtime.num_agents, "maxRuntimeSeconds": config.runtime.max_runtime_seconds, + "maxIterations": config.runtime.max_iterations, }, }, }