minor fixes

pytorch · Jul 30, 2024 · 07e4e1a · 07e4e1a
1 parent 90ad18b
commit 07e4e1a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 37 deletions.
diff --git a/docs/source/reference/envs.rst b/docs/source/reference/envs.rst
@@ -790,10 +790,10 @@ to be able to create this other composition:
     BurnInTransform
     CatFrames
     CatTensors
-    Crop
     CenterCrop
     ClipTransform
     Compose
+    Crop
     DTypeCastTransform
     DeviceCastTransform
     DiscreteActionProjection

diff --git a/test/test_transforms.py b/test/test_transforms.py
@@ -2178,18 +2178,8 @@ def test_transform_no_env(self, keys, h, nchannels, batch, device):
                 assert observation_spec[key].shape == torch.Size([nchannels, 20, h])
 
     @pytest.mark.parametrize("nchannels", [3])
-    @pytest.mark.parametrize(
-        "batch",
-        [
-            [2]
-        ]
-    )
-    @pytest.mark.parametrize(
-        "h",
-        [
-            None
-        ]
-    )
+    @pytest.mark.parametrize("batch", [[2]])
+    @pytest.mark.parametrize("h", [None])
     @pytest.mark.parametrize("keys", [["observation_pixels"]])
     @pytest.mark.parametrize("device", get_default_devices())
     def test_transform_model(self, keys, h, nchannels, batch, device):
@@ -2214,18 +2204,8 @@ def test_transform_model(self, keys, h, nchannels, batch, device):
         assert (td.get("dont touch") == dont_touch).all()
 
     @pytest.mark.parametrize("nchannels", [3])
-    @pytest.mark.parametrize(
-        "batch",
-        [
-            [2]
-        ]
-    )
-    @pytest.mark.parametrize(
-        "h",
-        [
-            None
-        ]
-    )
+    @pytest.mark.parametrize("batch", [[2]])
+    @pytest.mark.parametrize("h", [None])
     @pytest.mark.parametrize("keys", [["observation_pixels"]])
     @pytest.mark.parametrize("device", get_default_devices())
     def test_transform_compose(self, keys, h, nchannels, batch, device):
@@ -2254,18 +2234,8 @@ def test_transform_compose(self, keys, h, nchannels, batch, device):
         assert (tdc.get("dont touch") == dont_touch).all()
 
     @pytest.mark.parametrize("nchannels", [3])
-    @pytest.mark.parametrize(
-        "batch",
-        [
-            [2]
-        ]
-    )
-    @pytest.mark.parametrize(
-        "h",
-        [
-            None
-        ]
-    )
+    @pytest.mark.parametrize("batch", [[2]])
+    @pytest.mark.parametrize("h", [None])
     @pytest.mark.parametrize("keys", [["observation_pixels"]])
     @pytest.mark.parametrize("rbclass", [ReplayBuffer, TensorDictReplayBuffer])
     def test_transform_rb(