[Test] Fix Neuron installation and execution for trainium test.

Author: gmarciani

tests/integration-tests/tests/trainium/test_trainium/test_trainium/neuron-ccl.sh

@@ -6,7 +6,7 @@
 
 cat <<'EOF' >submission-script.sh
 #!/bin/bash
-set -x
+set -xe
 
 # FIXME remove this repo once packages are public available
 TEMPORARY_ARTIFACTS_BUCKET_PATH=s3://aws-parallelcluster-beta/neuron/
@@ -52,7 +52,7 @@ TOTAL_RANK=$(($SLURM_NNODES*32))
 # python3 inst-sweep/genneffs_nccl.py -n <total-number-of-ranks> --all --output <output-dir>
 NEFF_FILE=test_nccl_64r_50allg_int8_393216/0/file.neff
 if [[ ! -f $NEFF_FILE ]]; then
-  aws s3 cp ${TEMPORARY_ARTIFACTS_BUCKET_PATH}test_nccl_64r_50allg_int8_393216_0_file.neff $NEFF_FILE
+  aws s3 cp ${TEMPORARY_ARTIFACTS_BUCKET_PATH}test_nccl_64r_50allg_int8_393216_0_file.neff $NEFF_FILE --region us-east-1
 fi
 
 # Print eth0 ip

tests/integration-tests/tests/trainium/test_trainium/test_trainium/neuron-installation.sh

@@ -1,4 +1,5 @@
 #!/bin/bash
+set -ex
 
 # Private Repository Access
 # I manually created a TrainiumPreviewRepository secret and TrainiumPreviewPolicy on 447714826191 account to permit access to Secret below
@@ -115,7 +116,7 @@ EOF
 
 function main() {
   # Download packages from S3 --> FIXME they should be installed from configured repository
-  aws s3 cp ${TEMPORARY_ARTIFACTS_BUCKET_PATH} . --recursive
+  aws s3 cp ${TEMPORARY_ARTIFACTS_BUCKET_PATH} . --recursive --region us-east-1
 
   local OS="$(grep "^ID=" /etc/os-release | cut -d"=" -f 2 | xargs)"
   case ${OS} in