Update worker

new_experiment/worker_pipeline.py

@@ -40,6 +40,8 @@ def process_message(body: bytes):
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument("--prefetch_count")
+    parser.add_argument("--queue_name")
     args = parser.parse_args()
-    process_queue('asr_benchmark_experiments', process_message,
+    queue_name = 'asr_benchmark_experiments' if args.prefetch_count in [None, ''] else args.prefetch_count
+    process_queue(queue_name, process_message,
                   1 if args.prefetch_count in [None, ''] else int(args.prefetch_count))