amend

pytorch · Jul 26, 2024 · e049f47 · e049f47
1 parent 4b3b09e
commit e049f47
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/torchrl/modules/tensordict_module/rnn.py b/torchrl/modules/tensordict_module/rnn.py
@@ -543,7 +543,6 @@ def make_tensordict_primer(self):
             >>> from tensordict.nn import TensorDictSequential as Seq, TensorDictModule as Mod
             >>>
             >>> env = TransformedEnv(GymEnv("Pendulum-v1"), InitTracker())
-            >>> assert env.base_env.batch_locked
             >>> lstm_module = LSTMModule(
             ...     input_size=env.observation_spec["observation"].shape[-1],
             ...     hidden_size=64,
@@ -1294,7 +1293,6 @@ def make_tensordict_primer(self):
             >>> from tensordict.nn import TensorDictSequential as Seq, TensorDictModule as Mod
             >>>
             >>> env = TransformedEnv(GymEnv("Pendulum-v1"), InitTracker())
-            >>> assert env.base_env.batch_locked
             >>> gru_module = GRUModule(
             ...     input_size=env.observation_spec["observation"].shape[-1],
             ...     hidden_size=64,