Skip to content

Commit cf8d572

Browse files
committed
fix: type comparisons
1 parent 7611491 commit cf8d572

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pydra_ml/report.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -97,9 +97,9 @@ def plot_summary(summary, output_dir=None, filename="shap_plot", plot_top_n_shap
9797
# plot without all bootstrapping values
9898
summary = summary[["mean", "std", "min", "max"]]
9999
num_features = len(list(summary.index))
100-
if (plot_top_n_shap != 1 and type(plot_top_n_shap) == float) or type(
100+
if (plot_top_n_shap != 1 and type(plot_top_n_shap) is float) or type(
101101
plot_top_n_shap
102-
) == int:
102+
) is int:
103103
# if plot_top_n_shap != 1.0 but includes 1 (int)
104104
if plot_top_n_shap <= 0:
105105
raise ValueError(

0 commit comments

Comments
 (0)