aboutsummaryrefslogtreecommitdiffstats
path: root/benchmark_just_do_work.py
diff options
context:
space:
mode:
authorJan Tuomi <jan@jantuomi.fi>2025-05-15 12:34:57 +0300
committerJan Tuomi <jan@jantuomi.fi>2025-05-15 12:34:57 +0300
commitae36f44de7b9adbfa56525dd0046311ea7181288 (patch)
tree2cf714e28240204dd95e5b35bcde608d6f135d65 /benchmark_just_do_work.py
parent8d4fa3b038493df8f2b6108cef669d5434b1736e (diff)
Run formatter, add ruff config
Diffstat (limited to 'benchmark_just_do_work.py')
-rw-r--r--benchmark_just_do_work.py10
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)