diff --git a/src/transformers/benchmark/benchmark_utils_generic.py b/src/transformers/benchmark/benchmark_utils_generic.py index 8cb5a4da8c..fcf36ff927 100644 --- a/src/transformers/benchmark/benchmark_utils_generic.py +++ b/src/transformers/benchmark/benchmark_utils_generic.py @@ -102,6 +102,7 @@ class BenchMark: class SpeedBenchMark(BenchMark): + """A simple class used to benchmark the running time of a callable.""" def _measure(self, func, number=3, repeat=1): self._buffer["measure_kwargs"]["number"] = number self._buffer["measure_kwargs"]["repeat"] = repeat