Merge pull request #348 from dirac-institute/dep_fix

Fix deprecation warning
dirac-institute · Sep 27, 2023 · 95701ed · 95701ed
2 parents 17fd2b5 + 8241902
commit 95701ed
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/kbmod/file_utils.py b/src/kbmod/file_utils.py
@@ -221,13 +221,13 @@ def trajectory_from_np_object(result):
             The corresponding trajectory object.
         """
         trj = kb.Trajectory()
-        trj.x = int(result["x"])
-        trj.y = int(result["y"])
-        trj.vx = float(result["vx"])
-        trj.vy = float(result["vy"])
-        trj.flux = float(result["flux"])
-        trj.lh = float(result["lh"])
-        trj.obs_count = int(result["num_obs"])
+        trj.x = int(result["x"][0])
+        trj.y = int(result["y"][0])
+        trj.vx = float(result["vx"][0])
+        trj.vy = float(result["vy"][0])
+        trj.flux = float(result["flux"][0])
+        trj.lh = float(result["lh"][0])
+        trj.obs_count = int(result["num_obs"][0])
         return trj
 
     @staticmethod