requirements_cpu.txt

#
# python  (3.8.5)
# pip     (21.1.1)
#
# author: @contimatteo
#

#############################
## Dev

pylint
yapf

#############################
## Not-Versioned

numpy
python-dotenv
keras-rl2
scikit-learn
wandb

#############################
## Versioned

keras==2.3.1

# tensorflow-probability==0.10.0
tensorflow==2.2.2

# https://discourse.aicrowd.com/t/unable-to-install-flatland/1391/7
importlib-metadata==0.23

flatland-rl==2.2.2

#############################

# OSX Big Sur compatibility
# (https://github.com/openai/gym/issues/2101)
# pyglet==1.5.11