Add an assertion check on terminals and timeouts

takuseno · May 6, 2024 · c1560bf · c1560bf
1 parent b734d4e
commit c1560bf
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/d3rlpy/dataset/episode_generator.py b/d3rlpy/dataset/episode_generator.py
@@ -62,6 +62,10 @@ def __init__(
         assert (
             np.sum(np.logical_and(terminals, timeouts)) == 0
         ), "terminals and timeouts never become True at the same time"
+        assert (np.sum(terminals) + np.sum(timeouts)) > 0, (
+            "No episode termination was found. Either terminals"
+            " or timeouts must include non-zero values."
+        )
 
         self._observations = observations
         self._actions = actions