amend

Author: vmoens

torchrl/envs/llm/datasets/ifeval.py

@@ -7,7 +7,7 @@
 from typing import Any, Callable, Literal
 
 import torch
-from tensordict import lazy_stack, TensorClass, TensorDict
+from tensordict import lazy_stack, TensorClass, TensorDict, NonTensorStack, NonTensorData
 from torchrl._utils import logger as torchrl_logger
 from torchrl.data import Composite, NonTensor, Unbounded
 from torchrl.envs import StepCounter

torchrl/envs/transforms/transforms.py

@@ -6525,7 +6525,7 @@ def _reset_func(
                     # FIXME: use masked op
                     # tensordict_reset = tensordict_reset.clone()
                     reset_val = self.default_value(reset=_reset)
-                    # This is safe because env.reset calls _update_during_reset which will discard the new data
+                    # This is safE because env.reset calls _update_during_reset which will discard the new data
                     # tensordict_reset = (
                     #     self.container.full_observation_spec.zero().select(
                     #         *reset_val.keys(True)