security_vision/Trajectron-plus-plus
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix compatibility with numpy > 1.20

Browse source
This commit is contained in:
Ruben van de Ven 2023-10-04 21:16:02 +02:00
parent d182787c83
commit f4d860907c
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
trajectron/environment/scene_graph.py
View file

@ -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: