From 82f27125a2848905dbc225042ce5bee599d08953 Mon Sep 17 00:00:00 2001 From: Ahmad Sharif Date: Tue, 12 Nov 2024 13:37:38 -0800 Subject: [PATCH] . --- benchmarks/decoders/benchmark_decoders_library.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/benchmarks/decoders/benchmark_decoders_library.py b/benchmarks/decoders/benchmark_decoders_library.py index 14ca1da1..ae07727f 100644 --- a/benchmarks/decoders/benchmark_decoders_library.py +++ b/benchmarks/decoders/benchmark_decoders_library.py @@ -3,9 +3,9 @@ import subprocess import urllib.request from concurrent.futures import ThreadPoolExecutor, wait +from dataclasses import dataclass from itertools import product from pathlib import Path -from typing import NamedTuple import matplotlib.pyplot as plt import numpy as np @@ -480,7 +480,8 @@ def get_metadata(video_file_path: str) -> VideoStreamMetadata: return VideoDecoder(video_file_path).metadata -class BatchParameters(NamedTuple): +@dataclass +class BatchParameters: num_threads: int batch_size: int