Skip to content

Commit 62ee42d

Browse files
authored
Avoid warning in tests due to torch deprecation (#749)
1 parent 103f714 commit 62ee42d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/test_frame_dataclasses.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def test_framebatch_indexing():
139139
assert isinstance(fb_fancy, FrameBatch)
140140
assert fb_fancy.data.shape == (3, N, C, H, W)
141141

142-
fb_fancy = fb[[[0], [1]]] # select T=0 and N=1.
142+
fb_fancy = fb[([0], [1])] # select T=0 and N=1.
143143
assert isinstance(fb_fancy, FrameBatch)
144144
assert fb_fancy.data.shape == (1, C, H, W)
145145

0 commit comments

Comments
 (0)