[FIX] Update python in conda env, adapt doctest to numpy>=2

f-dangel · Sep 18, 2024 · 3cd7b2a · 3cd7b2a
1 parent b740fbc
commit 3cd7b2a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.conda_env.yml b/.conda_env.yml
@@ -3,7 +3,7 @@ channels:
   - pytorch
   - defaults
 dependencies:
-  - pip=21.2.4
-  - python=3.8.5
+  - python=3.9.16
+  - pip=23.1.2
   - pip:
     - -e .[lint,test,docs]
diff --git a/curvlinops/diagonal/hutchinson.py b/curvlinops/diagonal/hutchinson.py
@@ -50,7 +50,7 @@ class HutchinsonDiagonalEstimator:
         >>> error_high_precision = norm(diag_A - diag_A_high_precision)
         >>> assert error_low_precision > error_high_precision
         >>> round(error_low_precision, 4), round(error_high_precision, 4)
-        (5.7268, 0.1525)
+        (np.float64(5.7268), np.float64(0.1525))
     """
 
     def __init__(self, A: LinearOperator):