From 0fa4f99e35826f7967c5c6dcba28eef2c10c8c1e Mon Sep 17 00:00:00 2001 From: gmega Date: Fri, 20 Dec 2024 19:49:07 -0300 Subject: [PATCH] chore: run formatter --- benchmarks/core/utils.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/benchmarks/core/utils.py b/benchmarks/core/utils.py index ce914f9..92fde77 100644 --- a/benchmarks/core/utils.py +++ b/benchmarks/core/utils.py @@ -48,7 +48,7 @@ class RandomTempData(ExperimentData[TInitialMetadata]): def await_predicate( - predicate: Callable[[], bool], timeout: float = 0, polling_interval: float = 0 + predicate: Callable[[], bool], timeout: float = 0, polling_interval: float = 0 ) -> bool: current = time() while (timeout == 0) or ((time() - current) <= timeout): @@ -79,7 +79,9 @@ def megabytes(n: int) -> int: @contextmanager -def temp_random_file(size: int, name: str = "data.bin", batch_size: int = megabytes(50)): +def temp_random_file( + size: int, name: str = "data.bin", batch_size: int = megabytes(50) +): with tempfile.TemporaryDirectory() as temp_dir_str: temp_dir = Path(temp_dir_str) random_file = temp_dir / name