From 3b61364e387b5f40f55b4b3d90a14e17e455784c Mon Sep 17 00:00:00 2001
From: Eddie Goynes <eggoynes@amazon.com>
Date: Fri, 18 Nov 2022 10:17:48 -0800
Subject: [PATCH] Updates for v2.3.1

---
 CHANGELOG.md                   | 5 +++++
 source/install-gpu-drivers.ps1 | 4 ++--
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index f39ff03..1a944c4 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -5,6 +5,11 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [2.3.1] - 2022-11-18
+### Fixed 
+- Update Nvidia Driver installer for Cloud Editing hosts script updated to pull GPU drivers from us-east-1 bucket location. This fixes regions outside of us-east-1 that were unable to install drivers for hosts provisioned outside of us-east-1
+
+
 ## [2.3.0] - 2022-06-09
 
 - Update the Domain Join process to use Systems Manager Document rather than powershell [Github Issue #7]
diff --git a/source/install-gpu-drivers.ps1 b/source/install-gpu-drivers.ps1
index 00b5ff2..922e75d 100644
--- a/source/install-gpu-drivers.ps1
+++ b/source/install-gpu-drivers.ps1
@@ -31,7 +31,7 @@ function downloadGpuDrivers($Bucket, $BucketPrefix, $OsMatchStr, $DownloadDir) {
     Write-Host "Download from $Bucket and match on $OsMatchStr"
     $DriverFilename = '<not found>'
     # https://docs.aws.amazon.com/powershell/latest/reference/TOC.html
-    $Objects = Get-S3Object -BucketName $Bucket -Prefix $BucketPrefix
+    $Objects = Get-S3Object -BucketName $Bucket -Prefix $BucketPrefix -Region us-east-1
     foreach ($Object in $Objects) {
         Write-Host $Object.Key
         $FileName = $Object.Key
@@ -42,7 +42,7 @@ function downloadGpuDrivers($Bucket, $BucketPrefix, $OsMatchStr, $DownloadDir) {
         if ($FileName -ne '' -and $Object.Size -ne 0 -and
            ($FileName -match $OsMatchStr -or $FileName -match 'LicenseAgreement')) {
 		    $FullFilePath = Join-Path $DownloadDir $FileName
-	        Copy-S3Object -BucketName $Bucket -Key $Object.Key -LocalFile $FullFilePath
+	        Copy-S3Object -BucketName $Bucket -Key $Object.Key -LocalFile $FullFilePath -Region us-east-1
             Write-Host "Copying file $FileName"
             Write-Host "LocalFileName: $LocalFileName"
             Write-Host "FullFilePath: $FullFilePath"