From d0d8224957f605f2960d7cf7f03cb8431b53c219 Mon Sep 17 00:00:00 2001
From: Yunsong Zhou <75066007+ZhouYunsong-SJTU@users.noreply.github.com>
Date: Mon, 19 Feb 2024 16:30:56 -0800
Subject: [PATCH] fix a bug in nusc converter (#60)

---
 scenarionet/converter/nuscenes/utils.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/scenarionet/converter/nuscenes/utils.py b/scenarionet/converter/nuscenes/utils.py
index 9ef9d93..5873f9d 100644
--- a/scenarionet/converter/nuscenes/utils.py
+++ b/scenarionet/converter/nuscenes/utils.py
@@ -474,15 +474,15 @@ def convert_nuscenes_scenario(
 
     # No traffic light in nuscenes at this stage
     result[SD.DYNAMIC_MAP_STATES] = {}
-    track_to_predict = result[SD.TRACKS][instance_token]
-    result[SD.METADATA]["tracks_to_predict"] = {
-        instance_token: {
-            "track_index": list(result[SD.TRACKS].keys()).index(instance_token),
-            "track_id": instance_token,
-            "difficulty": 0,
-            "object_type": track_to_predict['type']
-        }
-    }
+    # track_to_predict = result[SD.TRACKS][instance_token]
+    # result[SD.METADATA]["tracks_to_predict"] = {
+    #     instance_token: {
+    #         "track_index": list(result[SD.TRACKS].keys()).index(instance_token),
+    #         "track_id": instance_token,
+    #         "difficulty": 0,
+    #         "object_type": track_to_predict['type']
+    #     }
+    # }
     # map
     result[SD.MAP_FEATURES] = get_map_features(scene_info, nuscenes, map_center, map_radius, only_lane=only_lane)
     # add back map center