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