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

Summary: Pull Request resolved: pytorch#1387 Differential Revision: D64008689
igorsugak · Oct 18, 2024 · 5db0ee8 · 5db0ee8
1 parent aef19a4
commit 5db0ee8
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]