diff --git a/requirements.txt b/requirements.txt
index b428e399..d5f6079b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,5 @@
+gym
+shimmy
 pycryptodome>=3.9.4
 requests>=2.21.0
 PyYAML>=5.3.1
@@ -19,4 +21,4 @@ websockets>=8.1
 torch>=2.0.1
 torchvision>=0.15.2
 stable-baselines3>=1.4.0
-RPi.GPIO
\ No newline at end of file
+RPi.GPIO