dayhoff-tools 1.1.30__py3-none-any.whl → 1.1.32__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 = {

dayhoff_tools/deployment/job_runner.py

@@ -94,9 +94,9 @@ def run_command() -> None:
             stderr=None,  # Use parent's stderr
         )
 
-        logger.info("Command completed successfully")
+        logger.info("Job command completed successfully")
     except subprocess.CalledProcessError as e:
-        logger.error(f"Command failed with return code: {e.returncode}")
+        logger.error(f"Job command failed with return code: {e.returncode}")
         raise
     except Exception as e:
         logger.error(f"Error executing command: {str(e)}")
@@ -153,8 +153,6 @@ def run_job(
         if mode in ["execute", "setup_and_execute"]:
             run_command()
 
-        logger.info("Job completed successfully")
-
     except Exception as e:
         logger.error(f"Job failed with error: {str(e)}", exc_info=True)
         sys.exit(1)

{dayhoff_tools-1.1.30.dist-info → dayhoff_tools-1.1.32.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dayhoff-tools
-Version: 1.1.30
+Version: 1.1.32
 Summary: Common tools for all the repos at Dayhoff Labs
 Author: Daniel Martin-Alarcon
 Author-email: dma@dayhofflabs.com
@@ -38,6 +38,7 @@ Requires-Dist: tqdm (>=4.67.1) ; extra == "full"
 Requires-Dist: transformers (==4.36.2) ; extra == "full"
 Requires-Dist: transformers (>=4.36.2) ; extra == "embedders"
 Requires-Dist: typer (>=0.9.0)
+Requires-Dist: tzdata (>=2025.2)
 Description-Content-Type: text/markdown
 
 # dayhoff-tools

{dayhoff_tools-1.1.30.dist-info → dayhoff_tools-1.1.32.dist-info}/RECORD

@@ -8,9 +8,9 @@ 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=GFaRl3gVPQZcrJXw6lWQPJA-vZ6Bl2cDcEyzT9v8fK0,6188
 dayhoff_tools/deployment/deploy_utils.py,sha256=StFwbqnr2_FWiKVg3xnJF4kagTHzndqqDkpaIOaAn_4,26027
-dayhoff_tools/deployment/job_runner.py,sha256=vCiWNF0wE1Syszt66KAAa1cPTe3k96pqvUOWOwBkpoI,5129
+dayhoff_tools/deployment/job_runner.py,sha256=hljvFpH2Bw96uYyUup5Ths72PZRL_X27KxlYzBMgguo,5086
 dayhoff_tools/deployment/processors.py,sha256=q2io07xO6f6twEH1iLz9GFoGaKh76qC4kcv519Q2G7g,34583
 dayhoff_tools/deployment/swarm.py,sha256=MGcS2_x4RNFtnVjWlU_SwNfhICz8NlGYr9cYBK4ZKDA,21688
 dayhoff_tools/embedders.py,sha256=svP_ksm3FdyVZ8i8R9R5uoGu2qI_hVQ_eztG0drXkN8,36477
@@ -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.30.dist-info/METADATA,sha256=MliGVbaukSohfzKI0kc5i3ExD3kOBa5X1mBfXAG39C8,2810
-dayhoff_tools-1.1.30.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-dayhoff_tools-1.1.30.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
-dayhoff_tools-1.1.30.dist-info/RECORD,,
+dayhoff_tools-1.1.32.dist-info/METADATA,sha256=ySjustGiiVOM1hnk9npS0uZ0svx3vb2vy7oM1Z1sOa4,2843
+dayhoff_tools-1.1.32.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+dayhoff_tools-1.1.32.dist-info/entry_points.txt,sha256=iAf4jteNqW3cJm6CO6czLxjW3vxYKsyGLZ8WGmxamSc,49
+dayhoff_tools-1.1.32.dist-info/RECORD,,