From df009dba0e0688822d45b2c17b660eff0fc79695 Mon Sep 17 00:00:00 2001
From: Per Held <per.held@arm.com>
Date: Wed, 5 Nov 2025 12:42:56 +0100
Subject: [PATCH] Arm backend: Fix mypy warnings in test/models

Add ignores for third-party modules and change import common into
using the correct path.

Signed-off-by: per.held@arm.com
Change-Id: I0b1cfa4c5b7a1625b72c6cfa59da8f427a25a204
---
 backends/arm/test/models/test_conformer.py        | 2 +-
 backends/arm/test/models/test_deit_tiny_arm.py    | 9 ++++++---
 backends/arm/test/models/test_inception_v3_arm.py | 5 +++--
 backends/arm/test/models/test_mobilenet_v3_arm.py | 5 +++--
 backends/arm/test/models/test_resnet18.py         | 5 ++++-
 backends/arm/test/models/test_w2l_arm.py          | 2 +-
 6 files changed, 18 insertions(+), 10 deletions(-)

diff --git a/backends/arm/test/models/test_conformer.py b/backends/arm/test/models/test_conformer.py
index 6302528e4ae..f5a4c8c5053 100644
--- a/backends/arm/test/models/test_conformer.py
+++ b/backends/arm/test/models/test_conformer.py
@@ -18,7 +18,7 @@
     VgfPipeline,
 )
 
-from torchaudio.models import Conformer
+from torchaudio.models import Conformer  # type: ignore[import-untyped]
 
 input_t = Tuple[torch.Tensor, torch.IntTensor]  # Input x, y
 
diff --git a/backends/arm/test/models/test_deit_tiny_arm.py b/backends/arm/test/models/test_deit_tiny_arm.py
index 22685a079bd..b26c6b634c2 100644
--- a/backends/arm/test/models/test_deit_tiny_arm.py
+++ b/backends/arm/test/models/test_deit_tiny_arm.py
@@ -7,7 +7,7 @@
 
 from typing import Tuple
 
-import timm
+import timm  # type: ignore[import-untyped]
 
 import torch
 
@@ -19,8 +19,11 @@
     VgfPipeline,
 )
 
-from timm.data import IMAGENET_INCEPTION_MEAN, IMAGENET_INCEPTION_STD
-from torchvision import transforms
+from timm.data import (  # type: ignore[import-untyped]
+    IMAGENET_INCEPTION_MEAN,
+    IMAGENET_INCEPTION_STD,
+)
+from torchvision import transforms  # type: ignore[import-untyped]
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
diff --git a/backends/arm/test/models/test_inception_v3_arm.py b/backends/arm/test/models/test_inception_v3_arm.py
index 2cb180a87ea..13dfac3199f 100644
--- a/backends/arm/test/models/test_inception_v3_arm.py
+++ b/backends/arm/test/models/test_inception_v3_arm.py
@@ -5,11 +5,12 @@
 
 from typing import Tuple
 
-import common
 import pytest
 
 import torch
 
+from executorch.backends.arm.test import common
+
 from executorch.backends.arm.test.tester.test_pipeline import (
     EthosU55PipelineINT,
     EthosU85PipelineINT,
@@ -18,7 +19,7 @@
     VgfPipeline,
 )
 
-from torchvision import models, transforms
+from torchvision import models, transforms  # type: ignore[import-untyped]
 
 ic3 = models.inception_v3(weights=models.Inception_V3_Weights)
 ic3 = ic3.eval()
diff --git a/backends/arm/test/models/test_mobilenet_v3_arm.py b/backends/arm/test/models/test_mobilenet_v3_arm.py
index f3a8f27428b..0a9c5ba27fc 100644
--- a/backends/arm/test/models/test_mobilenet_v3_arm.py
+++ b/backends/arm/test/models/test_mobilenet_v3_arm.py
@@ -5,11 +5,12 @@
 
 from typing import Tuple
 
-import common
 import pytest
 
 import torch
 
+from executorch.backends.arm.test import common
+
 from executorch.backends.arm.test.tester.test_pipeline import (
     EthosU55PipelineINT,
     EthosU85PipelineINT,
@@ -18,7 +19,7 @@
     VgfPipeline,
 )
 
-from torchvision import models, transforms
+from torchvision import models, transforms  # type: ignore[import-untyped]
 
 mv3 = models.mobilenet_v3_small(weights=models.MobileNet_V3_Small_Weights)
 mv3 = mv3.eval()
diff --git a/backends/arm/test/models/test_resnet18.py b/backends/arm/test/models/test_resnet18.py
index 44abc1d34e1..3a40a3dfd06 100644
--- a/backends/arm/test/models/test_resnet18.py
+++ b/backends/arm/test/models/test_resnet18.py
@@ -17,7 +17,10 @@
 )
 
 from torchvision import transforms  # type: ignore[import-untyped]
-from torchvision.models import resnet18, ResNet18_Weights
+from torchvision.models import (  # type: ignore[import-untyped]
+    resnet18,
+    ResNet18_Weights,
+)
 
 model = resnet18(weights=ResNet18_Weights)
 model = model.eval()
diff --git a/backends/arm/test/models/test_w2l_arm.py b/backends/arm/test/models/test_w2l_arm.py
index d62d92f5fa2..0eda5f45875 100644
--- a/backends/arm/test/models/test_w2l_arm.py
+++ b/backends/arm/test/models/test_w2l_arm.py
@@ -20,7 +20,7 @@
     VgfPipeline,
 )
 
-from torchaudio import models
+from torchaudio import models  # type: ignore[import-untyped]
 
 input_t = Tuple[torch.Tensor]  # Input x