From 3a757c67f767ae10d22f9fecc622d13582ea5c31 Mon Sep 17 00:00:00 2001
From: Danny McCormick <dannymccormick@google.com>
Date: Wed, 10 Dec 2025 12:52:49 -0500
Subject: [PATCH] Add yaml to requirements

---
 sdks/python/container/common.gradle | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/sdks/python/container/common.gradle b/sdks/python/container/common.gradle
index ad64dbbb660b..0fb460d9d6b5 100644
--- a/sdks/python/container/common.gradle
+++ b/sdks/python/container/common.gradle
@@ -42,7 +42,7 @@ def generatePythonRequirements = tasks.register("generatePythonRequirements") {
               "${files(configurations.sdkSourceTarball.files).singleFile} " +
               "base_image_requirements.txt " +
               "container " +
-              "[gcp,dataframe,test,tfrecord] " +
+              "[gcp,dataframe,test,tfrecord,yaml] " +
               "${pipExtraOptions}"
     }
     // Generate versions for ML dependencies
@@ -53,7 +53,7 @@ def generatePythonRequirements = tasks.register("generatePythonRequirements") {
               "${files(configurations.sdkSourceTarball.files).singleFile} " +
               "base_image_requirements.txt " +
               "container/ml " +
-              "[gcp,dataframe,test,ml_cpu,tfrecord] " +
+              "[gcp,dataframe,test,ml_cpu,tfrecord,yaml] " +
               "${pipExtraOptions}"
     }
     // TODO(https://github.com/apache/beam/issues/36637)