Merge pull request #5248 from RasaHQ/pin-gym

tmbo · web-flow · commit 3d8fa7f6b4e4 · 2020-02-17T17:58:06.000+01:00
Pin gym
diff --git a/changelog/5231.bugfix.rst b/changelog/5231.bugfix.rst
@@ -0,0 +1 @@
+Fix segmentation fault when running ``rasa train`` or ``rasa shell``.
diff --git a/rasa/core/training/generator.py b/rasa/core/training/generator.py
@@ -1,7 +1,6 @@
 from collections import defaultdict, namedtuple, deque
 
 import copy
-import warnings
 import logging
 import random
 from tqdm import tqdm
diff --git a/requirements.txt b/requirements.txt
@@ -15,6 +15,9 @@ absl-py>=0.8.0
 setuptools >= 41.0.0
 tensorflow-probability==0.7.0
 tensor2tensor==1.14.0
+# fixes https://github.com/RasaHQ/rasa/issues/5231
+# remove when removing `tensor2tensor`
+gym==0.15.4
 apscheduler==3.6.0
 tqdm==4.31.0
 networkx==2.4.0
diff --git a/setup.py b/setup.py
@@ -46,6 +46,9 @@
     "setuptools >= 41.0.0",
     "tensorflow-probability~=0.7.0",
     "tensor2tensor~=1.14.0",
+    # fixes https://github.com/RasaHQ/rasa/issues/5231
+    # remove when removing `tensor2tensor`
+    "gym<=0.15.4",
     "apscheduler~=3.0",
     "tqdm~=4.0",
     "networkx~=2.4.0",

Original file line number	Diff line number	Diff line change
`@@ -0,0 +1 @@`
	`1`	+Fix segmentation fault when running ``rasa train`` or ``rasa shell``.