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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: dayhoff-tools
-Version: 1.1.30
+Version: 1.1.33
 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 → 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.30 → dayhoff_tools-1.1.33}/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 → dayhoff_tools-1.1.33}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "dayhoff-tools"
-version = "1.1.30"
+version = "1.1.33"
 description = "Common tools for all the repos at Dayhoff Labs"
 authors = [
     {name = "Daniel Martin-Alarcon", email = "dma@dayhofflabs.com"}
@@ -19,6 +19,7 @@ dependencies = [
     "typer>=0.9.0",
     "toml>=0.10",
     "questionary>=2.0.1",
+    "tzdata>=2025.2",
 ]
 requires-python = ">=3.10,<4.0"