dh-cli 0.2.7__tar.gz → 0.2.8__tar.gz

{dh_cli-0.2.7 → dh_cli-0.2.8}/PKG-INFO

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

{dh_cli-0.2.7 → dh_cli-0.2.8}/pyproject.toml

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

{dh_cli-0.2.7 → dh_cli-0.2.8}/src/dh_cli/batch/aws_batch.py

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

{dh_cli-0.2.7 → dh_cli-0.2.8}/src/dh_cli/batch/commands/submit.py

@@ -1,6 +1,7 @@
 """Submit command for generic batch jobs."""
 
 import os
+import shlex
 from pathlib import Path
 
 import click
@@ -179,8 +180,12 @@ def submit(
             job_name=job_id,
             job_definition=DEFAULT_JOB_DEFINITION,
             job_queue=job_queue,
+            command=shlex.split(job_command),
             array_size=job_array if job_array > 1 else None,
             environment=submit_env,
+            vcpus=job_vcpus,
+            memory_mb=_parse_memory(job_memory),
+            gpus=job_gpus,
             timeout_seconds=timeout_seconds,
             retry_attempts=job_retry,
             depends_on=depends_on,
@@ -210,6 +215,16 @@ def submit(
         raise SystemExit(1)
 
 
+def _parse_memory(value: str) -> int:
+    """Convert a memory string like ``30G`` or ``16384`` to MiB."""
+    v = value.strip().upper()
+    if v.endswith("G"):
+        return int(v[:-1]) * 1024
+    if v.endswith("M"):
+        return int(v[:-1])
+    return int(v)
+
+
 def _parse_timeout(timeout_str: str) -> int:
     """Parse timeout string to seconds.