[Minor] Capture error in CatFrame edit (#1498)

pytorch · Sep 6, 2023 · 72e8fa4 · 72e8fa4
1 parent d729f3d
commit 72e8fa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_transforms.py b/test/test_transforms.py
@@ -618,7 +618,7 @@ def test_transform_no_env(self, device, d, batch_size, dim, N):
         td = TensorDict(dict(zip(keys, key_tensors)), batch_size, device=device)
         if dim > 0:
             with pytest.raises(
-                ValueError, match="dim must be > 0 to accomodate for tensordict"
+                ValueError, match="dim must be < 0 to accomodate for tensordict"
             ):
                 cat_frames = CatFrames(N=N, in_keys=keys, dim=dim)
             return