diff --git a/generate_data.py b/generate_data.py index 865d06d..fd5924e 100644 --- a/generate_data.py +++ b/generate_data.py @@ -13,9 +13,19 @@ MAX_DISTANCE = 20_037.5 # Add argparse parser = argparse.ArgumentParser() -parser.add_argument("-c", "--country", help="Specify the country code", type=str, default="US") -parser.add_argument("-w", "--workers", help="Specify the number of workers", type=int, default=1) -parser.add_argument("-o", "--output-file", help="Specify the name of the output file (file.csv)", type=str, default="city_distances_full.csv") +parser.add_argument( + "-c", "--country", help="Specify the country code", type=str, default="US" +) +parser.add_argument( + "-w", "--workers", help="Specify the number of workers", type=int, default=1 +) +parser.add_argument( + "-o", + "--output-file", + help="Specify the name of the output file (file.csv)", + type=str, + default="city_distances_full.csv", +) args = parser.parse_args()