Skip to content

Commit

Permalink
add --num-threads argument to train cli (#5086)
Browse files Browse the repository at this point in the history
* add `--num-threads` argument to train cli

* Update rasa/cli/train.py

* Create 5086.feature.rst

* fixed tests

Co-authored-by: Tom Bocklisch <tom@rasa.com>
  • Loading branch information
mludv and tmbo committed May 18, 2020
1 parent 7c94557 commit 69f0409
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion test_rasa_train.py
Expand Up @@ -326,6 +326,7 @@ def test_train_help(run):
help_text = """usage: rasa train [-h] [-v] [-vv] [--quiet] [--data DATA [DATA ...]]
[-c CONFIG] [-d DOMAIN] [--out OUT]
[--augmentation AUGMENTATION] [--debug-plots]
[--num-threads NUM_THREADS]
[--fixed-model-name FIXED_MODEL_NAME] [--persist-nlu-data]
[--force]
{core,nlu} ..."""
Expand All @@ -340,7 +341,8 @@ def test_train_nlu_help(run: Callable[..., RunResult]):
output = run("train", "nlu", "--help")

help_text = """usage: rasa train nlu [-h] [-v] [-vv] [--quiet] [-c CONFIG] [--out OUT]
[-u NLU] [--fixed-model-name FIXED_MODEL_NAME]
[-u NLU] [--num-threads NUM_THREADS]
[--fixed-model-name FIXED_MODEL_NAME]
[--persist-nlu-data]"""

lines = help_text.split("\n")
Expand Down

0 comments on commit 69f0409

Please sign in to comment.