Add error handling and logging to NVIDIA repository setup

- Add error messages for unsupported OS versions and CPU architectures
- Change local readonly to readonly for nvidia repo paths to fix scope issues
- Add early returns when encountering unsupported configurations
- Improve TODO comments for better clarity
- Fix typo: "locatl" -> "local" in mariner script

Signed-off-by: Suraj Deshmukh <suraj.deshmukh@microsoft.com>

Author: surajssd

parts/linux/cloud-init/artifacts/mariner/cse_install_mariner.sh

@@ -170,11 +170,12 @@ installCredentialProviderFromPMC() {
 }
 
 updateDnfWithNvidiaPkg() {
-  local readonly nvidia_repo_path="/etc/yum.repos.d/nvidia-built-azurelinux.repo"
+  readonly nvidia_repo_path="/etc/yum.repos.d/nvidia-built-azurelinux.repo"
 
   if [ "$OS_VERSION" != "3.0" ]; then
-    # TODO
-    # Error out?
+    # TODO: Error out?
+    echo "NVIDIA repo setup is only supported on Azure Linux 3.0"
+    return
   fi
 
   local cpu_arch=$(getCPUArch) # Returns amd64 or arm64
@@ -185,8 +186,9 @@ updateDnfWithNvidiaPkg() {
   elif [ "$cpu_arch" = "arm64" ]; then
     repo_arch="sbsa"
   else
-    # TODO
-    # Error out?
+    # TODO: Error out?
+    echo "Unsupported CPU architecture: $cpu_arch"
+    return
   fi
 
   locatl nvidia_repo_url="https://developer.download.nvidia.com/compute/cuda/repos/azl3/${repo_arch}/cuda-azl3.repo"

parts/linux/cloud-init/artifacts/ubuntu/cse_install_ubuntu.sh

@@ -77,10 +77,10 @@ updateAptWithMicrosoftPkg() {
 }
 
 updateAptWithNvidiaPkg() {
-    local readonly nvidia_gpg_keyring_path="/etc/apt/keyrings/nvidia.pub"
+    readonly nvidia_gpg_keyring_path="/etc/apt/keyrings/nvidia.pub"
     mkdir -p "$(dirname "${nvidia_gpg_keyring_path}")"
 
-    local readonly nvidia_sources_list_path="/etc/apt/sources.list.d/nvidia.list"
+    readonly nvidia_sources_list_path="/etc/apt/sources.list.d/nvidia.list"
     local cpu_arch=$(getCPUArch)  # Returns amd64 or arm64
     local repo_arch=""
     local nvidia_ubuntu_release=""
@@ -90,17 +90,19 @@ updateAptWithNvidiaPkg() {
     elif [ "$cpu_arch" = "arm64" ]; then
         repo_arch="sbsa"
     else
-        # TODO
-        # Error out?
+        # TODO: Error out?
+        echo "Unknown CPU architecture: ${cpu_arch}"
+        return
     fi
 
     if [ "${UBUNTU_RELEASE}" = "22.04" ]; then
         nvidia_ubuntu_release="ubuntu2204"
     elif [ "${UBUNTU_RELEASE}" = "24.04" ]; then
         nvidia_ubuntu_release="ubuntu2404"
     else
-        # TODO
-        # Error out?
+        # TODO: Error out?
+        echo "Unsupported Ubuntu version for NVIDIA drivers: ${UBUNTU_RELEASE}"
+        return
     fi
 
     # Construct URLs based on detected architecture and Ubuntu version