diff --git a/setup.py b/setup.py index eca507e..342e5e4 100644 --- a/setup.py +++ b/setup.py @@ -115,7 +115,10 @@ def get_readme() -> str: "pytest-xdist", "pytype", "stable-baselines3>=0.9.0", - "pyglet>=1.4.0", + # TODO(adam): remove pyglet pin once Gym upgraded to >0.21 + # Workaround for https://github.com/openai/gym/issues/2986 + # Discussed in https://github.com/HumanCompatibleAI/imitation/pull/603 + "pyglet==1.5.27", "setuptools_scm~=7.0.5", *ATARI_REQUIRE, ]