Add benchmark deps

Author: scotts

.github/workflows/benchmarks.yaml

@@ -37,6 +37,9 @@ jobs:
           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
           ffmpeg -version
+      - name: Install benchmark dependencies
+        run: |
+          python -m pip install matplotlib pandas numpy
       - name: Build and install torchcodec
         run: |
           python -m pip install -e ".[dev]" --no-build-isolation -vvv
@@ -46,4 +49,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