diff --git a/rl_coach/core_types.py b/rl_coach/core_types.py index 33c7f67..6d6bfe4 100644 --- a/rl_coach/core_types.py +++ b/rl_coach/core_types.py @@ -638,6 +638,9 @@ class Episode(object): def length(self): return self._length + def __len__(self): + return self.length() + def get_transition(self, transition_idx): return self.transitions[transition_idx]