diff options
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) |