Make algorithm and bug_id required arguments for the similarity query script

This commit is contained in:
Marco Castelluccio 2019-10-03 20:52:24 +01:00
Родитель 0440fa1b2e
Коммит 6087e2df07
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -22,11 +22,11 @@ URL = "https://index.taskcluster.net/v1/task/project.relman.bugbug.train_similar
def parse_args(args): def parse_args(args):
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument( parser.add_argument(
"--algorithm", "algorithm",
help="Similarity algorithm to use", help="Similarity algorithm to use",
choices=similarity.model_name_to_class.keys(), choices=similarity.model_name_to_class.keys(),
) )
parser.add_argument("--bug_id", help="Bug to test") parser.add_argument("bug_id", help="Bug to test")
return parser.parse_args(args) return parser.parse_args(args)