huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add flag to process all tracks of a particular location

Browse Source
This commit is contained in:
Arec
2021-07-20 05:46:43 -07:00
parent 793bb64af4
commit 2e29f42d75
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/expert_data.py
View File

@@ -91,5 +91,11 @@ if __name__ == '__main__':
help='location (default 0)') help='location (default 0)')
parser.add_argument('--track', default=0, type=int, parser.add_argument('--track', default=0, type=int,
help='track number (default 0)') 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() args = parser.parse_args()
generate_expert_data(loc=args.loc,track=args.track) 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)