better docs

pytorch · Nov 1, 2024 · 82c12e4 · 82c12e4
1 parent 69bd3a3
commit 82c12e4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchrl/envs/transforms/transforms.py b/torchrl/envs/transforms/transforms.py
@@ -4594,9 +4594,9 @@ class TensorDictPrimer(Transform):
 
     When used in a `TransformedEnv`, the spec shapes must match the environment's shape if
     the parent environment is batch-locked (`env.batch_locked=True`). If the spec shapes and
-    parent shapes do not match, the spec shapes are modified in place to match the leading
-    dimensions of the parent's batch size.  This adjustment is made for cases where the primer
-    shape is known during instantiation, but the batch size is not known until reset.
+    parent shapes do not match, the spec shapes are modified in-place to match the leading
+    dimensions of the parent's batch size. This adjustment is made for cases where the parent
+    batch size dimension is not known during instantiation.
 
     Examples:
         >>> from torchrl.envs.libs.gym import GymEnv