fix: ruff

UTAustin-SwarmLab · Feb 6, 2024 · fa0c68e · fa0c68e
1 parent 34fe64f
commit fa0c68e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ns_vfs/data/frame.py b/ns_vfs/data/frame.py
@@ -128,5 +128,5 @@ def save_frames(self, path):
                     Image.fromarray(self.annotated_images[idx]).save(
                         f"{annotation_path}/{idx}_annotated.png"
                     )
-            except:
+            except:  # noqa: E722
                 pass
diff --git a/ns_vfs/model/vision/object_detection/clip_model.py b/ns_vfs/model/vision/object_detection/clip_model.py
@@ -86,8 +86,8 @@ def detect(self, frame_img: np.ndarray, classes: list) -> any:
                 .numpy()
             )
 
-            logits_per_image, logits_per_text = self.model(image, text)
-            probs = logits_per_text.softmax(dim=-1).cpu().numpy()
+            # logits_per_image, logits_per_text = self.model(image, text)
+            # probs = logits_per_text.softmax(dim=-1).cpu().numpy()
 
         self._labels = []
         if len(scores) > 0: