diff --git a/packages/prime/src/prime_cli/commands/env.py b/packages/prime/src/prime_cli/commands/env.py index 27719f69..0906d10c 100644 --- a/packages/prime/src/prime_cli/commands/env.py +++ b/packages/prime/src/prime_cli/commands/env.py @@ -1665,6 +1665,7 @@ def eval_env( verbose: bool = typer.Option(False, "--verbose", "-v", help="Verbose output"), save_results: bool = typer.Option(True, "--save-results", "-s", help="Save results to disk"), save_every: int = typer.Option(1, "--save-every", "-f", help="Save dataset every n rollouts"), + stream: bool = typer.Option(False, "--stream", help="Stream results as they complete"), save_to_hf_hub: bool = typer.Option(False, "--save-to-hf-hub", "-H", help="Save to HF Hub"), hf_hub_dataset_name: Optional[str] = typer.Option( None, "--hf-hub-dataset-name", "-D", help="HF Hub dataset name" @@ -1772,6 +1773,8 @@ def eval_env( cmd += ["-s"] if save_every is not None: cmd += ["-f", str(save_every)] + if stream: + cmd += ["--stream"] if save_to_hf_hub: cmd += ["-H"] if hf_hub_dataset_name: