Linting

una-auxme · Mar 10, 2024 · fd6ac66 · fd6ac66
1 parent d3237ec
commit fd6ac66
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/code/perception/src/vision_node.py b/code/perception/src/vision_node.py
@@ -394,10 +394,11 @@ def predict_ultralytics(self, image):
         transposed_image = np.transpose(cv_image, (2, 0, 1))
         image_np_with_detections = torch.tensor(transposed_image,
                                                 dtype=torch.uint8)
-        
+
        if 9 in output[0].boxes.cls:
-          asyncio.run(
-              self.process_traffic_lights(output[0], cv_image, image.header))
+          asyncio.run(self.process_traffic_lights(output[0],
+                                                  cv_image,
+                                                  image.header))
 
         c_boxes = torch.stack(c_boxes)
         # print(image_np_with_detections.shape, c_boxes.shape, c_labels)