From 2e29f42d7579d7bd8b28af63e757c5e2635f5483 Mon Sep 17 00:00:00 2001 From: Arec Date: Tue, 20 Jul 2021 05:46:43 -0700 Subject: [PATCH] add flag to process all tracks of a particular location --- src/expert_data.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/src/expert_data.py b/src/expert_data.py index 70921ce..0b47c35 100644 --- a/src/expert_data.py +++ b/src/expert_data.py @@ -91,5 +91,11 @@ if __name__ == '__main__': help='location (default 0)') parser.add_argument('--track', default=0, type=int, help='track number (default 0)') + parser.add_argument('--all-tracks', action='store_true', + help='whether to process all tracks at location') args = parser.parse_args() - generate_expert_data(loc=args.loc,track=args.track) \ No newline at end of file + if args.all_tracks: + for i in range(intersim.MAX_TRACKS): + generate_expert_data(loc=args.loc, track=i) + else: + generate_expert_data(loc=args.loc,track=args.track) \ No newline at end of file