diff --git a/python/unittest_py/requirements.txt b/python/unittest_py/requirements.txt
index fe8382faa0c34c600c7a228baac34803ecc1492e..9165764adcaf4cccd3e9ad32fea28ca883761af0 100644
--- a/python/unittest_py/requirements.txt
+++ b/python/unittest_py/requirements.txt
@@ -3,6 +3,7 @@ coverage
 pycrypto ; platform_system != "Windows"
 mock
 gym
+pygame==2.1.0
 hypothesis
 opencv-python<=4.2.0.32
 visualdl