Fix stepper division by zero

fixes #34
ecmwf-lab · Mar 6, 2024 · 08ef357 · 08ef357
1 parent c098534
commit 08ef357
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ai_models/stepper.py b/ai_models/stepper.py
@@ -41,6 +41,9 @@ def __call__(self, i, step):
         self.last = now
 
     def __exit__(self, *args):
+        if self.num_steps == 0:
+            return
+
         elapsed = time.time() - self.start
         LOG.info("Elapsed: %s.", seconds(elapsed))
         LOG.info("Average: %s per step.", seconds(elapsed / self.num_steps))