From 0603527bb2adb40d268b7980f12486e92f5dcfe3 Mon Sep 17 00:00:00 2001 From: Kokouvi <52327882+KingKokouvi@users.noreply.github.com> Date: Tue, 30 Jan 2024 11:12:54 +0000 Subject: [PATCH] Update inference_example.py fix formatting --- examples/traffic_analysis/inference_example.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/examples/traffic_analysis/inference_example.py b/examples/traffic_analysis/inference_example.py index 787db52b..baa98c00 100644 --- a/examples/traffic_analysis/inference_example.py +++ b/examples/traffic_analysis/inference_example.py @@ -51,7 +51,8 @@ class DetectionsManager: if len(detections_all) > 0: detections_all.class_id = np.vectorize( lambda x: self.tracker_id_to_zone_id.get(x, -1) - )(detections_all.tracker_id) + ) + (detections_all.tracker_id) else: detections_all.class_id = np.array([], dtype=int) return detections_all[detections_all.class_id != -1]