From f4d860907c13b137215061f9ef9c2b62aa5e61e1 Mon Sep 17 00:00:00 2001 From: Ruben van de Ven Date: Wed, 4 Oct 2023 21:16:02 +0200 Subject: [PATCH] Fix compatibility with numpy > 1.20 --- trajectron/environment/scene_graph.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/trajectron/environment/scene_graph.py b/trajectron/environment/scene_graph.py index 1113bd4..4d969e9 100644 --- a/trajectron/environment/scene_graph.py +++ b/trajectron/environment/scene_graph.py @@ -135,10 +135,10 @@ class TemporalSceneGraph(object): position_cube = np.full((total_timesteps, N, 2), np.nan) adj_cube = np.zeros((total_timesteps, N, N), dtype=np.int8) - dist_cube = np.zeros((total_timesteps, N, N), dtype=np.float) + dist_cube = np.zeros((total_timesteps, N, N), dtype=float) node_type_mat = np.zeros((N, N), dtype=np.int8) - node_attention_mat = np.zeros((N, N), dtype=np.float) + node_attention_mat = np.zeros((N, N), dtype=float) for node_idx, node in enumerate(nodes): if online: