Merge pull request #260 from EricFillion/ef/2-3-1

New Version: 2.3.1
EricFillion · Sep 11, 2021 · 68ef679 · 68ef679
2 parents 2768f7f + 5482381
commit 68ef679
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/docs/pages/1-text-generation/3-settings.md b/docs/pages/1-text-generation/3-settings.md
@@ -38,7 +38,7 @@ GENSettings() contains the  fields shown in Table 1.0
 | bad_words            | None  | List of words/phrases that cannot be generated.                            | 
 
 
-#### Examples 1.2:  
+#### Example 1.2:  
 
  ```python
 from happytransformer import HappyGeneration, GENSettings

diff --git a/setup.py b/setup.py
@@ -9,7 +9,7 @@
 setup(
     name = 'happytransformer',
     packages = find_packages(),
-    version = '2.3.0',
+    version = '2.3.1',
     license='Apache 2.0',
     description = "Happy Transformer is an API built on top of Hugging Face's Transformer library that makes it easy to utilize state-of-the-art NLP models.",
     long_description= readme,

diff --git a/tests/test_gen.py b/tests/test_gen.py
@@ -27,7 +27,8 @@ def test_default_min_max_length():
 def test_bad_words():
     happy_gen = HappyGeneration()
     # Test single words
-    args_test_word_single = GENSettings(bad_words=["new","tool"])
+    args_test_word_single = GENSettings(bad_words=["new", "tool"])
+    # result without bad_words:  "a new field of research that has been gaining momentum"
     output_single = happy_gen.generate_text("Artificial intelligence is ", args=args_test_word_single)
     output_words_single = output_single.text.split()
     for phrase in args_test_word_single.bad_words: