lint

Signed-off-by: xadupre <[email protected]>
xadupre · Dec 21, 2024 · 94dfd7a · 94dfd7a
1 parent 47dae5e
commit 94dfd7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxmltools/convert/coreml/shape_calculators/OneHotEncoder.py b/onnxmltools/convert/coreml/shape_calculators/OneHotEncoder.py
@@ -26,7 +26,7 @@ def calculate_one_hot_encoder_output_shapes(operator):
         operator.outputs[0].type = FloatTensorType(
             [N, len(int_categories)], doc_string=operator.outputs[0].type.doc_string
         )
-    elif len(str_categories) > 0 and type(operator.inputs[0].type) == StringTensorType:
+    elif len(str_categories) > 0 and type(operator.inputs[0].type) is StringTensorType:
         operator.outputs[0].type = FloatTensorType(
             [N, len(str_categories)], doc_string=operator.outputs[0].type.doc_string
         )