fixing all syha 2

AIStream-Peelout · Jun 24, 2024 · ff5cd9a · ff5cd9a
1 parent 2f1ea7a
commit ff5cd9a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/flood_forecast/explain_model_output.py b/flood_forecast/explain_model_output.py
@@ -235,7 +235,7 @@ def deep_explain_model_heatmap(
     # heatmap one prediction sequence at datetime_start
     # (seq_len*forecast_len) per fop feature
     to_explain = history
-    shap_values = deep_explainer.shap_values(to_explain)
+    shap_values = deep_explainer.shap_values(to_explain, check_additivity=False)
     shap_values = fix_shap_values(shap_values, history)
     shap_values = np.stack(shap_values)
     if len(shap_values.shape) != 4:

diff --git a/requirements.txt b/requirements.txt
@@ -12,7 +12,7 @@ google-cloud-storage
 plotly~=5.20.0
 pytz>=2022.1
 setuptools~=69.5.1
-numpy>=1.21
+numpy==1.26
 requests
 torchvision>=0.6.0
 mpld3>=0.5