Fix the doctest due to numpy 2.0 (#654)

Summary: Pull Request resolved: #654 Doctest fails due to numpy's upgrade into 2.0, which changes the way of printing. astropy/astropy#15095 Reviewed By: EnayatUllah Differential Revision: D58873298 fbshipit-source-id: 5c4a21ff590c70b9643a1d921bc32b258e215616
pytorch · Jun 27, 2024 · 1235e1e · 1235e1e
1 parent 3619619
commit 1235e1e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/opacus/accountants/analysis/rdp.py b/opacus/accountants/analysis/rdp.py
@@ -39,7 +39,7 @@
     ...     rdp += compute_rdp(q=q, noise_multiplier=sigma, steps=steps, orders=orders)
 
     >>> epsilon, opt_order = get_privacy_spent(orders=orders, rdp=rdp, delta=1e-5)
-    >>> epsilon, opt_order  # doctest: +NUMBER
+    >>> float(epsilon), int(opt_order)  # doctest: +NUMBER
     (0.336, 23)
 
 """

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,4 @@
-numpy>=1.15
+numpy>=1.15,<2.0
 torch>=2.0
 scipy>=1.2
 opt-einsum>=3.3.0