Merge pull request #9 from NVIDIA/sshUser

Enable AWS ENV VAR key auth
NVIDIA · Feb 14, 2024 · a7bd07b · a7bd07b
2 parents 227eeed + 8d9ac35
commit a7bd07b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/provider/aws/aws.go b/pkg/provider/aws/aws.go
@@ -83,7 +83,11 @@ type Client struct {
 
 func New(log *logger.FunLogger, env v1alpha1.Environment, cacheFile string) (*Client, error) {
 	// Create an AWS session and configure the EC2 client
-	cfg, err := config.LoadDefaultConfig(context.TODO(), config.WithRegion(env.Spec.Region))
+	region := env.Spec.Region
+	if envRegion := os.Getenv("AWS_REGION"); envRegion != "" {
+		region = envRegion
+	}
+	cfg, err := config.LoadDefaultConfig(context.TODO(), config.WithRegion(region))
 	if err != nil {
 		return nil, err
 	}