update to numpy bool_ output

tianluyuan · Jun 18, 2024 · b4997bf · b4997bf
1 parent 3f6b986
commit b4997bf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/sphere/distribution/distribution.py b/sphere/distribution/distribution.py
@@ -605,8 +605,8 @@ def normalize(self, return_num_iterations=False):
         >>> gamma2 = np.array([0.0, 1.0, 0.0])
         >>> gamma3 = np.array([0.0, 0.0, 1.0])
         >>> tiny = FB8Distribution.minimum_value_for_kappa
-        >>> abs(fb82(gamma1, gamma2, gamma3, tiny, 0.0).normalize() - 4*np.pi) < 4*np.pi*1E-12
-        True
+        >>> np.abs(fb82(gamma1, gamma2, gamma3, tiny, 0.0).normalize() - 4*np.pi) < 4*np.pi*1E-12
+        np.True_
         >>> for kappa in [0.01, 0.1, 0.2, 0.5, 2, 4, 8, 16]:
         ...     print(np.abs(fb82(gamma1, gamma2, gamma3, kappa, 0.0).normalize() - 4*np.pi*np.sinh(kappa)/kappa) < 1E-15*4*np.pi*np.sinh(kappa)/kappa, end=' ')
         ...