dh-cli 0.2.3__tar.gz → 0.2.4__tar.gz

{dh_cli-0.2.3 → dh_cli-0.2.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dh-cli
-Version: 0.2.3
+Version: 0.2.4
 Summary: Dayhoff Labs developer CLI
 Author-email: Dayhoff Labs <dev@dayhofflabs.com>
 License: # PolyForm Noncommercial License 1.0.0

{dh_cli-0.2.3 → dh_cli-0.2.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "dh-cli"
-version = "0.2.3"
+version = "0.2.4"
 description = "Dayhoff Labs developer CLI"
 requires-python = ">=3.11"
 readme = "README.md"

{dh_cli-0.2.3 → dh_cli-0.2.4}/src/dh_cli/batch/aws_batch.py

@@ -83,6 +83,7 @@ class BatchClient:
         vcpus: int | None = None,
         memory_mb: int | None = None,
         gpus: int | None = None,
+        image_override: str | None = None,
     ) -> str:
         """Submit a job to AWS Batch.
 
@@ -123,6 +124,8 @@ class BatchClient:
                 resource_requirements.append({"type": "GPU", "value": str(gpus)})
 
             container_overrides: dict[str, Any] = {}
+            if image_override:
+                container_overrides["image"] = image_override
             if environment:
                 container_overrides["environment"] = [
                     {"name": k, "value": v} for k, v in environment.items()

{dh_cli-0.2.3 → dh_cli-0.2.4}/src/dh_cli/batch/commands/submit.py

@@ -177,7 +177,7 @@ def submit(
 
         batch_job_id = client.submit_job(
             job_name=job_id,
-            job_definition=job_image or DEFAULT_JOB_DEFINITION,
+            job_definition=DEFAULT_JOB_DEFINITION,
             job_queue=job_queue,
             array_size=job_array if job_array > 1 else None,
             environment=submit_env,
@@ -185,6 +185,7 @@ def submit(
             retry_attempts=job_retry,
             depends_on=depends_on,
             share_identifier=get_aws_username(),
+            image_override=job_image,
         )
 
         # Update manifest with Batch job ID