diff --git a/.github/workflows/python.yaml b/.github/workflows/python.yaml
index 1c7cc8d3..28e4d4a2 100644
--- a/.github/workflows/python.yaml
+++ b/.github/workflows/python.yaml
@@ -61,7 +61,7 @@ jobs:
           cpplint --recursive cpp
       - name: cmake-format
         run: |
-          pip install cmakelang
+          pip install cmakelang[YAML]
           find . -name "CMakeLists.txt" -not -path "*third_party/*" | xargs cmake-format -c .cmake-format.yaml --check
 
   build_test_job: