diff --git a/compiler/setup.py b/compiler/setup.py
index 980499b30..7c6c4d38b 100644
--- a/compiler/setup.py
+++ b/compiler/setup.py
@@ -56,7 +56,7 @@ setuptools.setup(
         where=build_dir() + "tools/zamalang/python_packages/zamalang_core",
         include=["mlir", "mlir.*"],
     ),
-    install_requires=["numpy"],
+    install_requires=["numpy", "PyYAML"],
     package_dir={"": build_dir() + "tools/zamalang/python_packages/zamalang_core"},
     include_package_data=True,
     package_data={"": ["*.so"]},