fix nan

Signed-off-by: xadupre <[email protected]>
onnx · Dec 20, 2024 · 62b24b5 · 62b24b5
1 parent 51130ec
commit 62b24b5
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/skl2onnx/operator_converters/power_transformer.py b/skl2onnx/operator_converters/power_transformer.py
@@ -23,6 +23,12 @@
 )
 
 
+def get_nan():
+    if hasattr(np, "nan"):
+        return np.nan
+    return np.NAN
+
+
 def convert_powertransformer(
     scope: Scope, operator: Operator, container: ModelComponentContainer
 ):
@@ -78,7 +84,7 @@ def convert_powertransformer(
         y_gr0 = OnnxImputer(
             y_gr0,
             imputed_value_floats=[0.0],
-            replaced_value_float=getattr(np, "nan", getattr(np, "NAN")),  # noqa: B009
+            replaced_value_float=get_nan(),
             op_version=opv,
         )
         y_gr0 = OnnxMul(y_gr0, greater_mask, op_version=opv)
@@ -104,7 +110,7 @@ def convert_powertransformer(
         y_le0 = OnnxImputer(
             y_le0,
             imputed_value_floats=[0.0],
-            replaced_value_float=getattr(np, "nan", getattr(np, "NAN")),  # noqa: B009
+            replaced_value_float=get_nan(),
             op_version=opv,
         )
         y_le0 = OnnxMul(y_le0, less_mask, op_version=opv)
@@ -130,7 +136,7 @@ def convert_powertransformer(
         y_gr0_l_eq0 = OnnxImputer(
             y_gr0_l_eq0,
             imputed_value_floats=[0.0],
-            replaced_value_float=getattr(np, "nan", getattr(np, "NAN")),  # noqa: B009
+            replaced_value_float=get_nan(),
             op_version=opv,
         )
         y_gr0_l_eq0 = OnnxMul(y_gr0_l_eq0, lambda_zero_mask, op_version=opv)