fix missed removed argument from cli

mllam · Oct 1, 2024 · dba94b3 · dba94b3
1 parent f3566b0
commit dba94b3
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/neural_lam/train_model.py b/neural_lam/train_model.py
@@ -253,9 +253,7 @@ def main(input_args=None):
 
     # Load model parameters Use new args for model
     ModelClass = MODELS[args.model]
-    model = ModelClass(
-        args, datastore=datastore, forcing_window_size=args.forcing_window_size
-    )
+    model = ModelClass(args, datastore=datastore)
 
     if args.eval:
         prefix = f"eval-{args.eval}-"