From c2a908f976cea3a275385cb9b46323f51d9e65e6 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Thu, 3 Jul 2025 13:08:09 +0100 Subject: [PATCH] Avoid warning in tests due to torch deprecation --- test/test_frame_dataclasses.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_frame_dataclasses.py b/test/test_frame_dataclasses.py index 73e06d208..003e49530 100644 --- a/test/test_frame_dataclasses.py +++ b/test/test_frame_dataclasses.py @@ -139,7 +139,7 @@ def test_framebatch_indexing(): assert isinstance(fb_fancy, FrameBatch) assert fb_fancy.data.shape == (3, N, C, H, W) - fb_fancy = fb[[[0], [1]]] # select T=0 and N=1. + fb_fancy = fb[([0], [1])] # select T=0 and N=1. assert isinstance(fb_fancy, FrameBatch) assert fb_fancy.data.shape == (1, C, H, W)