dayhoff-tools 1.1.31__py3-none-any.whl → 1.1.33__py3-none-any.whl

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.dist-info → dayhoff_tools-1.1.33.dist-info}/METADATA

@@ -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.dist-info → dayhoff_tools-1.1.33.dist-info}/RECORD

@@ -8,7 +8,7 @@ dayhoff_tools/cli/swarm_commands.py,sha256=5EyKj8yietvT5lfoz8Zx0iQvVaNgc3SJX1z2z
 dayhoff_tools/cli/utility_commands.py,sha256=ER4VrJt4hu904MwrcltUXjwBWT4uFrP-aPXjdXyT3F8,24685
 dayhoff_tools/deployment/base.py,sha256=8tXwsPYvRo-zV-aNhHw1c7Rji-KWg8S5xoCCznFnVVI,17412
 dayhoff_tools/deployment/deploy_aws.py,sha256=O0gQxHioSU_sNU8T8MD4wSOPvWc--V8eRRZzlRu035I,16446
-dayhoff_tools/deployment/deploy_gcp.py,sha256=DxBM4sUzwPK9RWLP9bSfr38n1HHl-TVrp4TsbdN8pUA,5795
+dayhoff_tools/deployment/deploy_gcp.py,sha256=s3nt-VsF0NrEA-J9Ire8G9zsFgtp8A6dHWT7IYQL5uQ,6529
 dayhoff_tools/deployment/deploy_utils.py,sha256=StFwbqnr2_FWiKVg3xnJF4kagTHzndqqDkpaIOaAn_4,26027
 dayhoff_tools/deployment/job_runner.py,sha256=hljvFpH2Bw96uYyUup5Ths72PZRL_X27KxlYzBMgguo,5086
 dayhoff_tools/deployment/processors.py,sha256=q2io07xO6f6twEH1iLz9GFoGaKh76qC4kcv519Q2G7g,34583
@@ -26,7 +26,7 @@ dayhoff_tools/intake/uniprot.py,sha256=BZYJQF63OtPcBBnQ7_P9gulxzJtqyorgyuDiPeOJq
 dayhoff_tools/logs.py,sha256=DKdeP0k0kliRcilwvX0mUB2eipO5BdWUeHwh-VnsICs,838
 dayhoff_tools/sqlite.py,sha256=jV55ikF8VpTfeQqqlHSbY8OgfyfHj8zgHNpZjBLos_E,18672
 dayhoff_tools/warehouse.py,sha256=TqV8nex1AluNaL4JuXH5zuu9P7qmE89lSo6f_oViy6U,14965
-dayhoff_tools-1.1.31.dist-info/METADATA,sha256=16xeYUw5Vk8m3cvEudAPbzXBv_CWnAq3Hn5ihSbpnIo,2843
-dayhoff_tools-1.1.31.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.1.31.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.1.31.dist-info/RECORD,,
+dayhoff_tools-1.1.33.dist-info/METADATA,sha256=8v8Kj6e0GdUHhtDx3l1huf7t1YwcnA0vlbbaKWwpBAw,2843
+dayhoff_tools-1.1.33.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.1.33.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.1.33.dist-info/RECORD,,