Ensure that nedts attribute of sensor is set.

simonpf · Oct 13, 2023 · 95c6886 · 95c6886
1 parent c8d6373
commit 95c6886
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gprof_nn/bin/train.py b/gprof_nn/bin/train.py
@@ -360,7 +360,7 @@ def run(args):
         return 1
     if args.simulated_tbs:
         sensor.use_simulated_tbs = True
-        sensor.nedts = args.nedts
+    sensor.nedts = args.nedts
 
     variant = args.variant
     if variant.upper() not in ["1D", "3D", "SIM", "HR"]: