diff --git a/model/ns3gym/requirements.txt b/model/ns3gym/requirements.txt index 02ec117e15..6fa1f23276 100644 --- a/model/ns3gym/requirements.txt +++ b/model/ns3gym/requirements.txt @@ -1 +1,4 @@ pyzmq +numpy +protobuf +gym \ No newline at end of file diff --git a/model/ns3gym/setup.py b/model/ns3gym/setup.py index e48eeb97a0..e0ee83e2d7 100644 --- a/model/ns3gym/setup.py +++ b/model/ns3gym/setup.py @@ -27,6 +27,6 @@ def readme(): description='OpenAI Gym meets ns-3', long_description='OpenAI Gym meets ns-3', keywords='openAI gym, ML, RL, ns-3', - install_requires=['pyzmq', 'numpy', 'protobuf', 'protobuf3', 'gym'], + install_requires=['pyzmq', 'numpy', 'protobuf', 'gym'], extras_require={}, )