parser = argparse.ArgumentParser() parser.add_argument("--device", default="auto") args = parser.parse_args() devices = available_devices() selected_device = select_device(args.device, devices)