chore: fix ruff lint warnings

tattle-made · Sep 11, 2024 · 5f73898 · 5f73898
1 parent 8a6b007
commit 5f73898
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 4 deletions.
diff --git a/src/core/operators/test_classify_video_zero_shot.py b/src/core/operators/test_classify_video_zero_shot.py
@@ -16,7 +16,7 @@ def tearDownClass(cls):
         # delete config files
         pass
 
-    # @skip
+    @skip
     def test_sample_video_from_disk(self):
         video_path = VideoFactory.make_from_file_on_disk(
             r"core/operators/sample_data/sample-cat-video.mp4"

diff --git a/src/core/operators/test_vid_vec_rep_clip.py b/src/core/operators/test_vid_vec_rep_clip.py
@@ -16,7 +16,7 @@ def tearDownClass(cls):
         # delete config files
         pass
 
-    # @skip
+    @skip
     def test_sample_video_from_disk(self):
         video_path = VideoFactory.make_from_file_on_disk(
             r"core/operators/sample_data/sample-cat-video.mp4"

diff --git a/src/core/store/es_vec.py b/src/core/store/es_vec.py
@@ -84,7 +84,7 @@ def refresh(self):
             self.client.indices.refresh()
 
     def find(self, index_name, vec):
-        if type(vec) == np.ndarray:
+        if isinstance(vec, np.ndarray):
             vec = vec.tolist()
 
         calculation = ""

diff --git a/src/worker/clustering_media/clustering_media_worker.py b/src/worker/clustering_media/clustering_media_worker.py
@@ -3,7 +3,8 @@
 from core.operators import audio_vec_embedding_clap
 from core.operators import vid_vec_rep_clip, classify_video_zero_shot
 from core.operators import cluster_embeddings, dimension_reduction
-import requests, json
+import requests
+import json
 from core.models.media_factory import AudioFactory, VideoFactory
 from time import sleep