From ae36f44de7b9adbfa56525dd0046311ea7181288 Mon Sep 17 00:00:00 2001 From: Jan Tuomi Date: Thu, 15 May 2025 12:34:57 +0300 Subject: Run formatter, add ruff config --- benchmark_just_do_work.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) (limited to 'benchmark_just_do_work.py') diff --git a/benchmark_just_do_work.py b/benchmark_just_do_work.py index b9a7696..a29c26a 100644 --- a/benchmark_just_do_work.py +++ b/benchmark_just_do_work.py @@ -9,6 +9,7 @@ import argparse from femtoqueue import FemtoQueue + def benchmark_femtoqueue(data_dir: str, num_tasks: int = 1000): queue = FemtoQueue(data_dir=data_dir, node_id="node1") @@ -23,9 +24,14 @@ def benchmark_femtoqueue(data_dir: str, num_tasks: int = 1000): break queue.done(task) + if __name__ == "__main__": - parser = argparse.ArgumentParser(description="FemtoQueue: Just run some tasks for external benchmarking") - parser.add_argument("data_dir", type=str, help="Data directory. Won't be cleaned up!") + parser = argparse.ArgumentParser( + description="FemtoQueue: Just run some tasks for external benchmarking" + ) + parser.add_argument( + "data_dir", type=str, help="Data directory. Won't be cleaned up!" + ) args = parser.parse_args() benchmark_femtoqueue(args.data_dir) -- cgit v1.3