Fix to mode setter

bilby-dev · Nov 25, 2024 · 0cf2c7e · 0cf2c7e
1 parent 2fa2bc1
commit 0cf2c7e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bilby/core/prior/joint.py b/bilby/core/prior/joint.py
@@ -755,7 +755,8 @@ def mode(self):
 
     @mode.setter
     def mode(self, mode):
-        if not np.isdtype(np.asarray(mode).dtype, "integral"):
+        mode = np.asarray(mode)
+        if not mode.dtype == int:
             raise ValueError("The mode to set must have integral data type.")
         if np.any(mode >= self.nmodes) or np.any(mode < 0):
             raise ValueError("The value of mode cannot be higher than the number of modes or smaller than zero.")