def _on_input(self):
    """Return True if we're currently processing input, False if output."""
    cur_sequence_id, _ = self._cur_position
    return cur_sequence_id % 2 == 0
Return True if we're currently processing input, False if output.
trax/rl/envs/data_envs.py