# Fork of gym with more flexible utility functions.
gym @ git+https://www.github.com/<anonymous>/gym@easier_custom_spaces#egg=gym
# nngeometry module used in the EWC method
nngeometry @ git+https://github.com/oleksost/nngeometry.git#egg=nngeometry
# Temporary fix for issue#128
pyyaml!=5.4.*,>=5.1
simple_parsing>=0.0.12.post4
matplotlib
torch>=1.7
torchvision
scikit-learn
tqdm
continuum==1.0.8
# Only required for the current demo:
wandb
plotly
pandas
# Only for python < 3.8
singledispatchmethod;python_version<'3.8'
# Temporarily fix the pytorch lightning version (issue #134)
pytorch-lightning==1.1.8
pytorch-lightning-bolts==0.3.0
# Requirements for running tests:
pytest-timeout
pytest-xdist
pytest-xvfb # Prevents the gym popups from displaying during tests.
# Required for the RL methods
# TODO: Move this to an 'extras' in setup.py?
stable-baselines3>=0.11.1
pyvirtualdisplay
# Required for the synbols dataset to work. 
h5py
