dayhoff-tools 1.1.31__tar.gz → 1.1.33__tar.gz

{dayhoff_tools-1.1.31 → dayhoff_tools-1.1.33}/PKG-INFO

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

{dayhoff_tools-1.1.31 → dayhoff_tools-1.1.33}/dayhoff_tools/deployment/deploy_gcp.py

@@ -75,9 +75,20 @@ def create_batch_job_config(config: dict, image_uri: str) -> dict:
         "commands": ["-c", " ".join(entrypoint_command)],
     }
 
+    # Handle container options - both shared memory and any custom options
+    docker_options = []
+
     # Add shared memory option if specified
     if "shared_memory" in config.get("docker", {}):
-        container_config["options"] = f"--shm-size={config['docker']['shared_memory']}"
+        docker_options.append(f"--shm-size={config['docker']['shared_memory']}")
+
+    # Add any custom Docker options if specified
+    if "options" in config.get("docker", {}):
+        docker_options.append(config["docker"]["options"])
+
+    # Set the options field if any options were collected
+    if docker_options:
+        container_config["options"] = " ".join(docker_options)
 
     # Build the task group configuration
     task_group = {
@@ -123,6 +134,17 @@ def create_batch_job_config(config: dict, image_uri: str) -> dict:
         print("Machine Type:", policy.get("machineType", "Not specified"))
         print("Accelerators:", policy.get("accelerators", "Not specified"))
         print("Environment Variables:", list(env_vars.keys()))
+        if (
+            "runnables" in task_group["taskSpec"]
+            and task_group["taskSpec"]["runnables"]
+        ):
+            print(
+                "Container Options:",
+                task_group["taskSpec"]["runnables"][0]
+                .get("container", {})
+                .get("options", "Not specified"),
+            )
+
     except KeyError as e:
         print(f"Warning: Could not find {e} in configuration")
 

{dayhoff_tools-1.1.31 → dayhoff_tools-1.1.33}/pyproject.toml

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