From f48083535624b933e24d64556215bceb14804584 Mon Sep 17 00:00:00 2001
From: Adam Gleave <adam@gleave.me>
Date: Mon, 21 Nov 2022 17:48:57 -0800
Subject: [PATCH] Pin pyglet version to workaround bug

---
 setup.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

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,
 ]