wrap conditional around s3_upload.py script

geoschem · Dec 6, 2023 · 5f71205 · 5f71205
1 parent beb957d
commit 5f71205
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/run_imi.sh b/run_imi.sh
@@ -198,6 +198,8 @@ cp $ConfigFile "${RunDirs}/config_${RunName}.yml"
 
 # Upload output to S3 if specified
 cd $InversionPath
-python src/utilities/s3_upload.py $ConfigFile
+if "$S3Upload"; then
+    python src/utilities/s3_upload.py $ConfigFile
+fi
 
 exit 0
diff --git a/src/utilities/s3_upload.py b/src/utilities/s3_upload.py
@@ -81,10 +81,7 @@ def zip_and_upload_to_s3(file_paths, bucket_name, s3_key):
     config_path = sys.argv[1]
     with open(config_path, "r") as f:
         config = yaml.safe_load(f)
-
-    if config["S3Upload"] != "true":
-        sys.exit(0)
-
+
     bucket = extract_s3_part(config["S3UploadPath"], "bucket")
     key = extract_s3_part(config["S3UploadPath"], "key")