Upgrade fbcode/pytorch to Python Scientific Stack 2 (#1387)

Summary: Pull Request resolved: #1387 Reviewed By: ipazarbasi Differential Revision: D64008689 fbshipit-source-id: 57e74b1e8b1e6f1cd7ce3e1fa67fc0999743bdfe
pytorch · Oct 22, 2024 · 1ba8977 · 1ba8977
1 parent f61847f
commit 1ba8977
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/captum/attr/_utils/visualization.py b/captum/attr/_utils/visualization.py
@@ -75,6 +75,7 @@ def _cumulative_sum_threshold(
     sorted_vals = np.sort(values.flatten())
     cum_sums = np.cumsum(sorted_vals)
     threshold_id = np.where(cum_sums >= cum_sums[-1] * 0.01 * percentile)[0][0]
+    # pyre-fixme[7]: Expected `float` but got `ndarray[typing.Any, dtype[typing.Any]]`.
     return sorted_vals[threshold_id]