From a5b08ea7269666cb084e22122b103176bbfdaf50 Mon Sep 17 00:00:00 2001
From: gs-olive <113141689+gs-olive@users.noreply.github.com>
Date: Thu, 3 Aug 2023 10:44:23 -0700
Subject: [PATCH] chore: Add pyyaml import to GHA Docker

---
 .github/workflows/docker_builder.yml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/.github/workflows/docker_builder.yml b/.github/workflows/docker_builder.yml
index 71e7cb74d4..817bc87c82 100644
--- a/.github/workflows/docker_builder.yml
+++ b/.github/workflows/docker_builder.yml
@@ -39,6 +39,7 @@ jobs:
       env:
         DOCKER_TAG: ${{ env.DOCKER_REGISTRY }}/${{ env.CONTAINER_NAME }}
       run: |
+        python3 -m pip install pyyaml
         TRT_VERSION=$(python3 -c "import versions; versions.tensorrt_version()")
         echo "TRT VERSION = ${TRT_VERSION}"
         CUDNN_VERSION=$(python3 -c "import versions; versions.cudnn_version()")