Fix tests

marqo-ai · Oct 3, 2024 · cc02da4 · cc02da4
1 parent 82b246f
commit cc02da4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/marqo/tensor_search/streaming_media_processor.py b/src/marqo/tensor_search/streaming_media_processor.py
@@ -206,9 +206,9 @@ def fetch_audio_chunk(url: str, start_time: float, duration: float, output_file:
         ffmpeg_command = [
             'ffmpeg', '-y' # Enable overwrite
             '-v', 'error',  # Suppress warnings and other output
-            '-i', url,  # Input file
+            '-i', str(url),  # Input file
             '-ss', str(start_time),  # Start time
-            '-t', duration,  # Duration
+            '-t', str(duration),  # Duration
             '-acodec', 'pcm_s16le',  # Audio codec
             '-ar', '44100',  # Audio sample rate
             '-f', 'wav',  # Output format

diff --git a/tests/s2_inference/test_video_ffmpeg_decode.py b/tests/s2_inference/test_video_ffmpeg_decode.py
@@ -10,6 +10,8 @@
 class TestVideoFFmpegDecode(unittest.TestCase):
     def setUp(self):
         self.output_file = "/tmp/test.mp4"
+        if os.path.exists(self.output_file):
+            os.remove(self.output_file)
 
     def tearDown(self):
         if os.path.exists(self.output_file):