Only download the prefilled DB when training

This commit is contained in:
Marco Castelluccio 2019-01-02 15:00:09 +01:00
Родитель 35bd1ee961
Коммит dbf992f696
2 изменённых файлов: 1 добавлений и 3 удалений

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

@ -6,7 +6,7 @@ install:
- pip install -r test-requirements.txt
script:
- flake8
- python run.py --download --train --goal bug
- python run.py --train --goal bug
- python run.py --goal bug
- python -m pytest tests/test_*.py
- python setup.py sdist

2
run.py
Просмотреть файл

@ -7,7 +7,6 @@ import argparse
from bugbug import bugzilla
from bugbug import db
from bugbug import labels
from bugbug import repository # noqa
if __name__ == '__main__':
@ -38,7 +37,6 @@ if __name__ == '__main__':
if args.train:
db.download()
bugzilla.download_bugs(labels.get_all_bug_ids())
model = model_class(args.lemmatization)
model.train()