Fix support for env vars

WATonomous · Dec 6, 2024 · 47fab52 · 47fab52
1 parent a674827
commit 47fab52
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/agent.py b/src/agent.py
@@ -96,8 +96,8 @@ def __init__(self, bucket_config, repo_config, workspace_dir: str):
             name: boto3.resource(
                 "s3",
                 endpoint_url=config["endpoint_url"],
-                aws_access_key_id=config["access_key_id"] or os.environ[config["access_key_id_env_var"]],
-                aws_secret_access_key=config["secret_key"] or os.environ[config["secret_key_env_var"]],
+                aws_access_key_id=config.get("access_key_id") or os.environ[config["access_key_id_env_var"]],
+                aws_secret_access_key=config.get("secret_key") or os.environ[config["secret_key_env_var"]],
             ).Bucket(config["bucket_name"])
             for name, config in bucket_config.items()
         }