From a830df2909f672b23d7e30e1a8c2ed67f6404e54 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 25 Nov 2024 17:20:41 +0100 Subject: [PATCH] Fix `test_auto_backbone_timm_model_from_pretrained` (#34877) fix Co-authored-by: ydshieh --- tests/models/auto/test_modeling_auto.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/auto/test_modeling_auto.py b/tests/models/auto/test_modeling_auto.py index 95d7168983..5766b4adae 100644 --- a/tests/models/auto/test_modeling_auto.py +++ b/tests/models/auto/test_modeling_auto.py @@ -239,7 +239,7 @@ class AutoModelTest(unittest.TestCase): # Check kwargs are correctly passed to the backbone model = AutoBackbone.from_pretrained("resnet18", use_timm_backbone=True, out_indices=(-2, -1)) - self.assertEqual(model.out_indices, (-2, -1)) + self.assertEqual(model.out_indices, [-2, -1]) # Check out_features cannot be passed to Timm backbones with self.assertRaises(ValueError):