dayhoff-tools 1.10.3__tar.gz → 1.11.0__tar.gz

{dayhoff_tools-1.10.3 → dayhoff_tools-1.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dayhoff-tools
-Version: 1.10.3
+Version: 1.11.0
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com

{dayhoff_tools-1.10.3 → dayhoff_tools-1.11.0}/dayhoff_tools/deployment/deploy_aws.py

@@ -477,10 +477,21 @@ def submit_aws_batch_job(
     print(f"Setting retry attempts to {retry_attempts}")
     job_submit_args["retryStrategy"] = {"attempts": retry_attempts}
 
-    # Add tags if specified in config
+    # Automatically add User tag for cost tracking
+    username = os.getenv("LOCAL_USER", "unknown_user")
+    default_tags = {"User": username}
+
+    # Merge with any tags specified in config (config tags take precedence)
     if "tags" in aws_config:
-        job_submit_args["tags"] = aws_config["tags"]
-        print(f"Adding tags to batch job: {aws_config['tags']}")
+        tags = {**default_tags, **aws_config["tags"]}
+    else:
+        tags = default_tags
+
+    job_submit_args["tags"] = tags
+    job_submit_args["propagateTags"] = (
+        True  # Propagate tags to ECS tasks and EC2 instances
+    )
+    print(f"Adding tags to batch job: {tags}")
 
     # Submit the job using the session client
     response = batch.submit_job(**job_submit_args)

{dayhoff_tools-1.10.3 → dayhoff_tools-1.11.0}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "dayhoff-tools"
-version = "1.10.3"
+version = "1.11.0"
 description = "Common tools for all the repos at Dayhoff Labs"
 authors = [
     {name = "Daniel Martin-Alarcon", email = "dma@dayhofflabs.com"}