mm/citybert
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

linting

Browse Source
This commit is contained in:
mm 2023-05-05 03:41:59 +00:00
parent 96f0c3a313
commit 1e38ce04c9
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
generate_data.py
View File

@ -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()