Add benchmark deps

Author: scotts

.github/workflows/benchmarks.yaml

@@ -35,6 +35,7 @@ jobs:
           # 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
           python -m pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu
+          python -m pip install decord matplotlib pandas numpy
           conda install "ffmpeg=7.0.1" pkg-config -c conda-forge
           ffmpeg -version
       - name: Build and install torchcodec
@@ -46,4 +47,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