diff options
| author | Jan Tuomi <jan@jantuomi.fi> | 2025-05-15 12:34:57 +0300 |
|---|---|---|
| committer | Jan Tuomi <jan@jantuomi.fi> | 2025-05-15 12:34:57 +0300 |
| commit | ae36f44de7b9adbfa56525dd0046311ea7181288 (patch) | |
| tree | 2cf714e28240204dd95e5b35bcde608d6f135d65 /benchmark_just_do_work.py | |
| parent | 8d4fa3b038493df8f2b6108cef669d5434b1736e (diff) | |
Run formatter, add ruff config
Diffstat (limited to 'benchmark_just_do_work.py')
| -rw-r--r-- | benchmark_just_do_work.py | 10 |
1 files changed, 8 insertions, 2 deletions
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) |
