make more numpy-thonic

ludwig-ai · Jun 6, 2022 · 5ba51a4 · 5ba51a4
1 parent 9d25b27
commit 5ba51a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ludwig/features/binary_feature.py b/ludwig/features/binary_feature.py
@@ -363,7 +363,7 @@ def postprocess_predictions(
         if probabilities_col in result:
             result[probabilities_col] = backend.df_engine.map_objects(
                 result[probabilities_col],
-                lambda prob: np.array([1 - prob, prob]).astype(result[probabilities_col].dtype),
+                lambda prob: np.array([1 - prob, prob], dtype=result[probabilities_col].dtype),
             )
 
             false_col = f"{probabilities_col}_{class_names[0]}"