Add default num_ensemble_members

ecmwf-lab · Dec 9, 2024 · f598b90 · f598b90
1 parent fd5125a
commit f598b90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ai_models_gencast/model.py b/src/ai_models_gencast/model.py
@@ -321,7 +321,7 @@ def parse_model_args(self, args):
         import argparse
 
         parser = argparse.ArgumentParser("ai-models gencast")
-        parser.add_argument("--num_ensemble_members", type=int, help="Number of ensemble members to run", required=True)
+        parser.add_argument("--num_ensemble_members", type=int, help="Number of ensemble members to run", default=1)
         parser.add_argument("--use-an", action="store_true")
         parser.add_argument("--override-constants")
         return parser.parse_args(args)