Adds aws session tokens

Author: DavidGOrtega

iterative/resource_runner.go

@@ -232,6 +232,7 @@ func provisionerCode(d *schema.ResourceData) (string, error) {
 	data["instance_gpu"] = d.Get("instance_gpu").(string)
 	data["AWS_SECRET_ACCESS_KEY"] = os.Getenv("AWS_SECRET_ACCESS_KEY")
 	data["AWS_ACCESS_KEY_ID"] = os.Getenv("AWS_ACCESS_KEY_ID")
+	data["AWS_SESSION_TOKEN"] = os.Getenv("AWS_SESSION_TOKEN")
 	data["AZURE_CLIENT_ID"] = os.Getenv("AZURE_CLIENT_ID")
 	data["AZURE_CLIENT_SECRET"] = os.Getenv("AZURE_CLIENT_SECRET")
 	data["AZURE_SUBSCRIPTION_ID"] = os.Getenv("AZURE_SUBSCRIPTION_ID")
@@ -280,6 +281,7 @@ sudo bash -c 'cat << EOF > /usr/bin/cml.sh
 
 export AWS_SECRET_ACCESS_KEY={{.AWS_SECRET_ACCESS_KEY}}
 export AWS_ACCESS_KEY_ID={{.AWS_ACCESS_KEY_ID}}
+export AWS_SESSION_TOKEN={{.AWS_SESSION_TOKEN}}
 export AZURE_CLIENT_ID={{.AZURE_CLIENT_ID}}
 export AZURE_CLIENT_SECRET={{.AZURE_CLIENT_SECRET}}
 export AZURE_SUBSCRIPTION_ID={{.AZURE_SUBSCRIPTION_ID}}