Skip to content

Commit

Permalink
Update
Browse files Browse the repository at this point in the history
[ghstack-poisoned]
  • Loading branch information
vmoens committed Dec 6, 2024
1 parent 2c82726 commit 20037b4
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion torchrl/envs/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -3266,7 +3266,7 @@ def maybe_reset(self, tensordict: TensorDictBase) -> TensorDictBase:
"""
if self._simple_done:
done = tensordict._get_str("done", default=None)
any_done = done.any()
any_done = done is not None and done.any()
if any_done:
tensordict._set_str(
"_reset",
Expand Down

0 comments on commit 20037b4

Please sign in to comment.