From 4822b389a5be73a6f5d687fbfc8694fe9758b914 Mon Sep 17 00:00:00 2001 From: Sina Sajadmanesh Date: Thu, 30 Apr 2026 15:14:26 +0200 Subject: [PATCH] Fix conditional check for imageType feature type --- coremltools/models/ml_program/experimental/perf_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/coremltools/models/ml_program/experimental/perf_utils.py b/coremltools/models/ml_program/experimental/perf_utils.py index ce8123dd1..4113b451d 100644 --- a/coremltools/models/ml_program/experimental/perf_utils.py +++ b/coremltools/models/ml_program/experimental/perf_utils.py @@ -119,7 +119,7 @@ def _gen_random_feature_value( elif feature_type == "multiArrayType": return _gen_random_multiarray_value(type=type.multiArrayType) - elif type == "imageType": + elif feature_type == "imageType": return _gen_random_image_value(type=type.imageType) else: