tuple not list

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Jan 8, 2025 · 7a84698 · 7a84698
1 parent bfbb78f
commit 7a84698
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_backbones.py b/tests/test_backbones.py
@@ -120,7 +120,7 @@ def test_out_indices(model_name, input_224):
 def test_out_indices_non_divisible(model_name, input_non_divisible):
     out_indices = [2, 4, 8, 10]
     backbone = timm.create_model(model_name, pretrained=False, features_only=True, num_frames=NUM_FRAMES, out_indices=out_indices, padding='constant')
-    assert backbone.feature_info.out_indices == out_indices
+    assert backbone.feature_info.out_indices == tuple(out_indices)
 
     output = backbone(input_non_divisible)
     full_output = backbone.forward_features(input_non_divisible)