diff --git a/ml/requirements-all.txt b/ml/requirements-all.txt
index da1cf22..9563501 100644
--- a/ml/requirements-all.txt
+++ b/ml/requirements-all.txt
@@ -66,7 +66,7 @@ pytest-pep8==1.0.6
 pytest-pylint==0.14.0
 python-dateutil==2.7.5
 pytz==2018.9
-PyYAML==5.1
+PyYAML==5.4
 requests==2.22.0
 s3transfer==0.2.0
 scipy==1.2.0
diff --git a/ml/requirements-api.txt b/ml/requirements-api.txt
index 5e1ab1c..6725e7b 100644
--- a/ml/requirements-api.txt
+++ b/ml/requirements-api.txt
@@ -1,5 +1,5 @@
 jsonpickle==1.1
-PyYAML==5.1
+PyYAML==5.4
 
 boto3==1.9.139
 botocore==1.12.151
diff --git a/ml/requirements.txt b/ml/requirements.txt
index 535c305..b168870 100644
--- a/ml/requirements.txt
+++ b/ml/requirements.txt
@@ -9,7 +9,7 @@ scipy==1.2.0
 
 
 jsonpickle==1.1
-PyYAML==5.1
+PyYAML==5.4
 
 boto3==1.9.139
 botocore==1.12.151