From 1f5fa669575ddcd3fd0c1d8ead8471fb71c17459 Mon Sep 17 00:00:00 2001
From: Marco Castelluccio <mcastelluccio@mozilla.com>
Date: Tue, 20 Nov 2018 16:46:25 +0100
Subject: [PATCH] Add script to run the training

---
 run.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)
 create mode 100644 run.py

diff --git a/run.py b/run.py
new file mode 100644
index 00000000..e79fb21b
--- /dev/null
+++ b/run.py
@@ -0,0 +1,16 @@
+# -*- coding: utf-8 -*-
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this file,
+# You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import argparse
+
+from bugbug import train
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--lemmatization', help='Perform lemmatization (using spaCy)', action='store_true')
+    parser.add_argument('--model', nargs='?', help='Path where to store the model file')
+    args = parser.parse_args()
+
+    train.train(model=args.model, lemmatization=args.lemmatization)