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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dh-cli
-Version: 0.2.3
+Version: 0.2.6
 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.6}/pyproject.toml

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

{dh_cli-0.2.3 → dh_cli-0.2.6}/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.
 
@@ -104,6 +105,11 @@ class BatchClient:
             BatchError: If submission fails
         """
         try:
+            if image_override:
+                job_definition = self._register_image_override(
+                    job_definition, image_override
+                )
+
             submit_args: dict[str, Any] = {
                 "jobName": job_name,
                 "jobDefinition": job_definition,
@@ -153,6 +159,39 @@ class BatchClient:
         except ClientError as e:
             raise BatchError(f"Failed to submit job: {e}")
 
+    def _register_image_override(
+        self, base_definition: str, image: str
+    ) -> str:
+        """Register a new revision of a job definition with a custom image.
+
+        AWS Batch containerOverrides doesn't support an image field, so we
+        create a new revision of the base definition with only the image
+        changed, preserving all other containerProperties (roles, volumes, etc.).
+
+        Returns the ARN of the new revision.
+        """
+        resp = self.batch.describe_job_definitions(
+            jobDefinitionName=base_definition, status="ACTIVE"
+        )
+        definitions = resp.get("jobDefinitions", [])
+        if not definitions:
+            raise BatchError(
+                f"Job definition not found: {base_definition}"
+            )
+
+        latest = sorted(definitions, key=lambda d: d["revision"])[-1]
+        container_props = latest["containerProperties"].copy()
+        container_props["image"] = image
+
+        reg_resp = self.batch.register_job_definition(
+            jobDefinitionName=base_definition,
+            type=latest.get("type", "container"),
+            containerProperties=container_props,
+        )
+        arn = reg_resp["jobDefinitionArn"]
+        logger.info(f"Registered job definition {arn} with image {image}")
+        return arn
+
     def submit_array_job_with_indices(
         self,
         job_name: str,

{dh_cli-0.2.3 → dh_cli-0.2.6}/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