Skip to content

Commit f272aa3

Browse files
authored
Fix conditional check for imageType feature type (#2679)
1 parent a3b2074 commit f272aa3

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

coremltools/models/ml_program/experimental/perf_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def _gen_random_feature_value(
119119
elif feature_type == "multiArrayType":
120120
return _gen_random_multiarray_value(type=type.multiArrayType)
121121

122-
elif type == "imageType":
122+
elif feature_type == "imageType":
123123
return _gen_random_image_value(type=type.imageType)
124124

125125
else:

0 commit comments

Comments
 (0)