From bf3113806c35184ec07ff76b9501646cef8ac711 Mon Sep 17 00:00:00 2001 From: Per Held Date: Fri, 7 Nov 2025 09:04:41 +0100 Subject: [PATCH] Arm backend: Add ignores for missing imports in test/models/sta... Some missing imports for third-party libraries used in stable_diffusion had ignores added to them. Signed-off-by: per.held@arm.com Change-Id: Icd86edd39ae4541813bc84c07fdab90571da98c2 --- .../models/stable_diffusion/test_SD3Transformer2DModel.py | 4 +++- .../models/stable_diffusion/test_vae_AutoencoderKL.py | 8 ++++++-- 2 files changed, 9 insertions(+), 3 deletions(-) diff --git a/backends/arm/test/models/stable_diffusion/test_SD3Transformer2DModel.py b/backends/arm/test/models/stable_diffusion/test_SD3Transformer2DModel.py index 9506fe727db..3e1f19dd39c 100644 --- a/backends/arm/test/models/stable_diffusion/test_SD3Transformer2DModel.py +++ b/backends/arm/test/models/stable_diffusion/test_SD3Transformer2DModel.py @@ -7,7 +7,9 @@ from typing import Tuple import torch -from diffusers.models.transformers import SD3Transformer2DModel +from diffusers.models.transformers import ( # type: ignore[import-not-found] + SD3Transformer2DModel, +) from executorch.backends.arm.test import common from executorch.backends.arm.test.models.stable_diffusion.stable_diffusion_module_test_configs import ( diff --git a/backends/arm/test/models/stable_diffusion/test_vae_AutoencoderKL.py b/backends/arm/test/models/stable_diffusion/test_vae_AutoencoderKL.py index a3c3a018131..5d33576a817 100644 --- a/backends/arm/test/models/stable_diffusion/test_vae_AutoencoderKL.py +++ b/backends/arm/test/models/stable_diffusion/test_vae_AutoencoderKL.py @@ -7,8 +7,12 @@ from typing import Tuple import torch -from diffusers.models.autoencoders import AutoencoderKL -from diffusers.utils.testing_utils import floats_tensor +from diffusers.models.autoencoders import ( # type: ignore[import-not-found] + AutoencoderKL, +) +from diffusers.utils.testing_utils import ( # type: ignore[import-not-found] + floats_tensor, +) from executorch.backends.arm.test import common from executorch.backends.arm.test.models.stable_diffusion.stable_diffusion_module_test_configs import (