Fix deprecation warning

dirac-institute · Sep 26, 2023 · 859c23b · 859c23b
1 parent eeb62d5
commit 859c23b
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