Fix precommit

ThomasMeissnerDS · Aug 19, 2024 · dce9510 · dce9510
1 parent 750abf4
commit dce9510
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 12 deletions.
diff --git a/bluecast/blueprints/cast.py b/bluecast/blueprints/cast.py
@@ -449,12 +449,11 @@ def fit(self, df: pd.DataFrame, target_col: str) -> None:
             shap_waterfall_plot(
                 explainer, self.conf_training.shap_waterfall_indices, self.class_problem
             )
-            if self.conf_training.show_dependence_plots_of_top_n_features > 0:
-                shap_dependence_plots(
-                    shap_values,
-                    x_test,
-                    self.conf_training.show_dependence_plots_of_top_n_features,
-                )
+            shap_dependence_plots(
+                shap_values,
+                x_test,
+                self.conf_training.show_dependence_plots_of_top_n_features,
+            )
         self.prediction_mode = True
 
     def fit_eval(

diff --git a/bluecast/blueprints/cast_regression.py b/bluecast/blueprints/cast_regression.py
@@ -421,12 +421,11 @@ def fit(self, df: pd.DataFrame, target_col: str) -> None:
             shap_waterfall_plot(
                 explainer, self.conf_training.shap_waterfall_indices, self.class_problem
             )
-            if self.conf_training.show_dependence_plots_of_top_n_features > 0:
-                shap_dependence_plots(
-                    shap_values,
-                    x_test,
-                    self.conf_training.show_dependence_plots_of_top_n_features,
-                )
+            shap_dependence_plots(
+                shap_values,
+                x_test,
+                self.conf_training.show_dependence_plots_of_top_n_features,
+            )
         self.prediction_mode = True
 
     def fit_eval(

diff --git a/dist/bluecast-1.6.0-py3-none-any.whl b/dist/bluecast-1.6.0-py3-none-any.whl
diff --git a/dist/bluecast-1.6.0.tar.gz b/dist/bluecast-1.6.0.tar.gz