From 60a4c83b8744a2db6b94744b9625e37dc79854e7 Mon Sep 17 00:00:00 2001 From: Matthew Caseres Date: Thu, 5 Jun 2025 20:39:50 -0500 Subject: [PATCH] Update comment to match model choices --- containers/BasicTerm_ME_python/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/containers/BasicTerm_ME_python/main.py b/containers/BasicTerm_ME_python/main.py index ba13c04..9e9cc0f 100644 --- a/containers/BasicTerm_ME_python/main.py +++ b/containers/BasicTerm_ME_python/main.py @@ -3,7 +3,7 @@ def main(): parser = argparse.ArgumentParser(description="Term ME model runner") parser.add_argument("--multiplier", type=int, default=100, help="Multiplier for model points") - # add an argument that must be either "torch" or "jax" + # add an argument that must be either "torch_recursive" or "jax_iterative" parser.add_argument("--model", type=str, default="jax_iterative", choices=["torch_recursive", "jax_iterative"], help="Model to run") args = parser.parse_args()