Fix naming when loading model file (#180)

This commit is contained in:
John Giannelos 2019-02-18 16:57:01 +01:00 коммит произвёл Marco
Родитель 1c0e25a893
Коммит 24cacd673c
1 изменённых файлов: 4 добавлений и 1 удалений

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

@ -28,7 +28,10 @@ if __name__ == '__main__':
parser.add_argument('--generate-sheet', help='Perform evaluation on bugs from last week and generate a csv file', action='store_true')
args = parser.parse_args()
model_file_name = f'{args.goal}model'
model_file_name = '{}{}model'.format(
args.goal,
'' if args.classifier == 'default' else args.classifier
)
if args.goal == 'bug':
from bugbug.models.bug import BugModel