[aws|azure] do not override osDesc architecture in WithOS

DataDog · Dec 27, 2024 · 25efe45 · 25efe45
1 parent 1a62db6
commit 25efe45
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/scenarios/aws/ec2/vmargs.go b/scenarios/aws/ec2/vmargs.go
@@ -37,10 +37,9 @@ func buildArgs(options ...VMOption) (*vmArgs, error) {
 }
 
 // WithOS sets the OS
-// Architecture defaults to os.AMD64Arch
 // Version defaults to latest
 func WithOS(osDesc os.Descriptor) VMOption {
-	return WithOSArch(osDesc, os.AMD64Arch)
+	return WithOSArch(osDesc, osDesc.Architecture)
 }
 
 // WithArch set the architecture and the operating system.

diff --git a/scenarios/azure/compute/vmargs.go b/scenarios/azure/compute/vmargs.go
@@ -36,10 +36,9 @@ func buildArgs(options ...VMOption) (*vmArgs, error) {
 }
 
 // WithOS sets the OS
-// Architecture defaults to os.AMD64Arch
 // Version defaults to latest
 func WithOS(osDesc os.Descriptor) VMOption {
-	return WithOSArch(osDesc, os.AMD64Arch)
+	return WithOSArch(osDesc, osDesc.Architecture)
 }
 
 // WithArch set the architecture and the operating system.