cleanup

arangoml · Feb 9, 2024 · c1a846b · c1a846b
1 parent 037fa4d
commit c1a846b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/adbpyg_adapter/adapter.py b/adbpyg_adapter/adapter.py
@@ -917,9 +917,9 @@ def __process_adb_edge_df(
  np.array([from_n.to_numpy(), to_n.to_numpy()]), dtype=torch.int64
  )
 
- empty_tensor = torch.tensor([], dtype=torch.int64)
+ empty_tensor = tensor([], dtype=torch.int64)
  existing_edge_index = edge_data.get("edge_index", empty_tensor)
- edge_data.edge_index = torch.cat((existing_edge_index, edge_index), dim=1)
+ edge_data.edge_index = cat((existing_edge_index, edge_index), dim=1)
 
  # 7. Deal with invalid edges
  if torch.any(edge_data.edge_index == -1):