diff --git a/requirements.txt b/requirements.txt
index 4e781b4..e1553f4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -4,6 +4,6 @@ UMNN==1.0
 h5py
 networkx==2.4
 pandas
-PyYAML==5.3.1
+PyYAML==5.4
 scikit-learn