Upgrade fbcode/pytorch to Python Scientific Stack 2

Summary: X-link: pytorch/captum#1387 X-link: pytorch/botorch#2584 X-link: pytorch/audio#3845 Differential Revision: D64008689
igorsugak · Oct 18, 2024 · bbf372b · bbf372b
1 parent aea78b3
commit bbf372b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/opacus/accountants/analysis/prv/prvs.py b/opacus/accountants/analysis/prv/prvs.py
@@ -16,6 +16,7 @@
 from typing import Tuple
 
 import numpy as np
+import numpy.typing as npt
 from scipy import integrate
 from scipy.special import erfc
 
@@ -133,7 +134,7 @@ def mean(self) -> float:
 
 @dataclass
 class DiscretePRV:
-    pmf: np.ndarray
+    pmf: npt.NDArray
     domain: Domain
 
     def __len__(self) -> int: