pass doctests w latest numpy version

jbloomlab · Aug 30, 2024 · 72151b1 · 72151b1
1 parent bac8e77
commit 72151b1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/polyclonal/polyclonal.py b/polyclonal/polyclonal.py
@@ -451,7 +451,7 @@ class Polyclonal:
 
     >>> model_data.mutations
     ('M1C', 'G2A', 'A4K', 'A4L')
-    >>> dict(sorted(model_data.mutations_times_seen.items()))
+    >>> {key: int(val) for (key, val) in sorted(model_data.mutations_times_seen.items())}
     {'A4K': 4, 'A4L': 3, 'G2A': 6, 'M1C': 6}
 
     The activities are evenly spaced from 1 to 0, while the mutation escapes
@@ -1652,8 +1652,7 @@ def _scaled_pseudo_huber(delta, r, calc_grad=False):
         ...       lambda r: Polyclonal._scaled_pseudo_huber(2, r, False)[0],
         ...       lambda r: Polyclonal._scaled_pseudo_huber(2, r, True)[1],
         ...       [2])
-        >>> err < 1e-7
-        True
+        >>> assert err < 1e-7
 
         """
         if delta <= 0: