From 38b305273ecf386f18c195a1bf7aa5ee369b8330 Mon Sep 17 00:00:00 2001
From: Colm Talbot <colm.talbot@ligo.org>
Date: Mon, 13 May 2024 08:25:02 -0700
Subject: [PATCH] MAINT: fixes for numpy>=2

---
 bilby/core/utils/io.py | 4 ++--
 bilby/gw/prior.py      | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/bilby/core/utils/io.py b/bilby/core/utils/io.py
index b6e9b79c5..80590faf4 100644
--- a/bilby/core/utils/io.py
+++ b/bilby/core/utils/io.py
@@ -264,9 +264,9 @@ def encode_for_hdf5(key, item):
 
     if isinstance(item, np.int_):
         item = int(item)
-    elif isinstance(item, np.float_):
+    elif isinstance(item, np.float64):
         item = float(item)
-    elif isinstance(item, np.complex_):
+    elif isinstance(item, np.complex128):
         item = complex(item)
     if isinstance(item, np.ndarray):
         # Numpy's wide unicode strings are not supported by hdf5
diff --git a/bilby/gw/prior.py b/bilby/gw/prior.py
index 81dd2d557..5e9f670d5 100644
--- a/bilby/gw/prior.py
+++ b/bilby/gw/prior.py
@@ -1436,7 +1436,7 @@ class HealPixMapPriorDist(BaseJointPriorDist):
                 self.update_distance(int(round(val)))
                 dist_samples[i] = self.distance_icdf(dist_samp[i])
         if self.distance:
-            sample = np.row_stack([sample[:, 0], sample[:, 1], dist_samples])
+            sample = np.vstack([sample[:, 0], sample[:, 1], dist_samples])
         return sample.reshape((-1, self.num_vars))
 
     def update_distance(self, pix_idx):
-- 
GitLab