Fix area initialize response

inverted-ai · Nov 29, 2023 · 9910f6e · 9910f6e
1 parent c059a5b
commit 9910f6e
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/invertedai/utils.py b/invertedai/utils.py
@@ -500,11 +500,10 @@ def inside_fov(center: Point, initialize_fov: float, point: Point) -> bool:
             file_path = f"{birdview_path}-{(area_center.x, area_center.y)}.jpg"
             response.birdview.decode_and_save(file_path)
 
-    return invertedai.api.InitializeResponse(
-        recurrent_states=agent_rs,
-        agent_states=agent_states,
-        agent_attributes=agent_attributes)
-
+    response.recurrent_states=agent_rs
+    response.agent_states=agent_states
+    response.agent_attributes
+    return response
 
 class APITokenAuth(AuthBase):
     def __init__(self, api_token):