make code bit more succinct

2025-12-06 04:12:13 +00:00 · 2025-11-22 22:48:55 +01:00 · 2025-11-22 22:48:55 +01:00 · df9a644e24
commit df9a644e24
parent a33d04dca1
1 changed files with 1 additions and 3 deletions
--- a/tasks/common.py
+++ b/tasks/common.py
@ -46,9 +46,7 @@ class Task:
                    f"Using {num_ex} examples instead."
                )
                self.stop = num_ex
-            stop = self.stop
-        else:
-            stop = self.num_examples()
+        stop = self.num_examples() if self.stop is None else self.stop
        step = self.step
        span = stop - start
        num = (span + step - 1) // step # ceil_div(span, step)