diff --git a/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt
index c232cb2..ab7c886 100644
--- a/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt
+++ b/modules/ml-pipeline/src/pipeline/src/requirements/predictions/requirements.txt
@@ -3,3 +3,4 @@ boto3==1.28.17
 pandas==1.5.3
 scikit-learn==1.3.0
 pyarrow==13.0.0
+PyYAML==6.0.1