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

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

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

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

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

@@ -105,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,
@@ -124,8 +129,6 @@ 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()
@@ -156,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,