From 52913c9b165602d28146f833ad14576974c02c90 Mon Sep 17 00:00:00 2001 From: Igor Sugak Date: Thu, 17 Oct 2024 20:56:05 -0700 Subject: [PATCH] Upgrade fbcode/pytorch to Python Scientific Stack 2 (#680) Summary: X-link: https://github.com/pytorch/captum/pull/1387 X-link: https://github.com/pytorch/botorch/pull/2584 X-link: https://github.com/pytorch/audio/pull/3845 Differential Revision: D64008689 --- opacus/accountants/analysis/prv/prvs.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/opacus/accountants/analysis/prv/prvs.py b/opacus/accountants/analysis/prv/prvs.py index b4ffcf76..957ad336 100644 --- 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: