update CLI command mapping

Wikidata · Jan 10, 2019 · 265003f · 265003f
1 parent ae61ca0
commit 265003f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/soweego/linker/cli.py b/soweego/linker/cli.py
@@ -1,8 +1,10 @@
 import click
-from soweego.linker import linking_strategies
+
+from soweego.linker import linking_strategies, train
 
 CLI_COMMANDS = {
-    'baseline': linking_strategies.baseline
+    'baseline': linking_strategies.baseline,
+    'train': train.train_cli
 }