From f100ef8ff50475a510215fe04cf5e52289088f6c Mon Sep 17 00:00:00 2001
From: Karl W Schulz <karl.schulz@amd.com>
Date: Fri, 17 Nov 2023 15:29:29 -0600
Subject: [PATCH] Update cmake config with Fix for pyyaml package installation
 check

Signed-off-by: Karl W Schulz <karl.schulz@amd.com>
---
 CMakeLists.txt | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 0b9911598e..1f6ad8618e 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -78,8 +78,13 @@ message(STATUS "Checking for required Python package dependencies...")
 set_property(GLOBAL PROPERTY pythonDepsFlag "groovy")
 
 function(checkPythonPackage [package])
+    # mapping for non-default package names
+    set(PACKAGE ${ARGV0})
+    if(${ARGV0} STREQUAL "pyyaml")
+        set(PACKAGE "yaml")
+    endif()
     execute_process(
-        COMMAND ${Python3_EXECUTABLE} -c "import ${ARGV0}"
+        COMMAND ${Python3_EXECUTABLE} -c "import ${PACKAGE}"
         OUTPUT_QUIET ERROR_QUIET
         RESULT_VARIABLE EXIT_CODE)
     if(${EXIT_CODE} EQUAL 0)