Add benchmark deps

.github/workflows/benchmarks.yaml

@@ -32,10 +32,9 @@ jobs:
         run: python -m pip install --upgrade pip
       - name: Install dependencies and FFmpeg
         run: |
-          # TODO: torchvision and torchaudio shouldn't be needed. They were only added
-          #  to silence an error as seen in https://github.com/pytorch/torchcodec/issues/203
+          conda install ffmpeg pkg-config -c conda-forge
           python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
-          conda install "ffmpeg=7.0.1" pkg-config -c conda-forge
+          python -m pip install decord matplotlib pandas numpy
           ffmpeg -version
       - name: Build and install torchcodec
         run: |
@@ -46,4 +45,3 @@ jobs:
       - name: TEST README data geeneration benchmark
         run: |
           python benchmarks/decoders/generate_readme_data.py --test_run
-

benchmarks/decoders/generate_readme_data.py

@@ -58,7 +58,7 @@ def main() -> None:
         patterns = ["mandelbrot"]
         fpses = [30]
         gop_sizes = [20]
-        durations = [10] # if this goes too low, we hit EOF errors in some decoders
+        durations = [10]  # if this goes too low, we hit EOF errors in some decoders
         pix_fmts = ["yuv420p"]
         ffmpeg_path = "ffmpeg"
         min_runtime_seconds = 1