Only tensorflow v. 1 supported?

[nbecker]

In a first test I created a new venv, then pip install -r requirements.txt,
then python ./train_dqn_agent.py.
I get:
ModuleNotFoundError: No module named 'tensorflow.contrib'

Does this mean only tf 1.x will work? requirements.txt does not specify tf version, I got 2.5.0

[aldebaro]

Thanks for your feedback. We will check and report here.

[aldebaro]

Sorry, but Stable-Baselines 2 restricts us to use tf 1.x. Based on your contribution, we have updated the readme to inform:
Please note that the Stable-Baselines 2 documentation (https://stable-baselines.readthedocs.io/en/master/guide/install.html) states that ``Stable-Baselines supports Tensorflow versions from 1.8.0 to 1.15.0, and does not work on Tensorflow versions 2.0.0 and above. PyTorch support is done in Stable-Baselines3''. Hence, you need to adopt Tensorflow 1.x when executing the RL code.

[cleversonahum]

It can also be a problem with the python version you are using @nbecker , since tensorflow 1.X is not available for Python versions >3.7 (https://pypi.org/project/tensorflow/1.15.5/). For instance, it seems to work using Python 3.7 and installing tensoflow 1.14.0. I've just sent a pull request #2 that adds this information in the README and adds the tensorflow version into requirements file.