From 72e8fa491380bcd9d6e5b0a5da778db5d8ee982f Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Wed, 6 Sep 2023 13:25:28 +0100 Subject: [PATCH] [Minor] Capture error in CatFrame edit (#1498) --- test/test_transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_transforms.py b/test/test_transforms.py index b84133b0e42..088b667e847 100644 --- 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