From ff5cd9a8168fdeda6782d48105e8702585f962cf Mon Sep 17 00:00:00 2001 From: isaacmg Date: Mon, 24 Jun 2024 17:36:09 -0400 Subject: [PATCH] fixing all syha 2 --- flood_forecast/explain_model_output.py | 2 +- requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/flood_forecast/explain_model_output.py b/flood_forecast/explain_model_output.py index 134a276a9..cfbe2ecff 100644 --- 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 index b3273e1e9..cd0ff50d3 100644 --- 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