diff --git a/mlir/python/requirements.txt b/mlir/python/requirements.txt
index aaf480f0bdc72f..16a3dd39c7fcfe 100644
--- a/mlir/python/requirements.txt
+++ b/mlir/python/requirements.txt
@@ -1,4 +1,4 @@
 numpy>=1.19.5, <=1.23.5
 pybind11>=2.8.0, <=2.10.3
-PyYAML==6.0
+PyYAML>= 5.3.1, <=6.0
 dataclasses>=0.6, <=0.8