diff --git a/pyproject.toml b/pyproject.toml
index d41c804395773a6f17c6db3f52bed78e81bf1740..55adff0787a505fb34e5f844ddae3460e1f31ae5 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -11,6 +11,7 @@ python = ">=3.10,<3.12"
 stable-baselines3 = "^2.3.2"
 gymnasium = "^1.0.0"
 scipy = "^1.14.1"
+pyyaml = "^6.0.2"
 
 
 [build-system]