runnable 0.30.4__tar.gz → 0.30.5__tar.gz

{runnable-0.30.4 → runnable-0.30.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: runnable
-Version: 0.30.4
+Version: 0.30.5
 Summary: Add your description here
 Author-email: "Vammi, Vijay" <vijay.vammi@astrazeneca.com>
 License-File: LICENSE

{runnable-0.30.4 → runnable-0.30.5}/extensions/nodes/torch.py

@@ -6,7 +6,7 @@ import string
 from datetime import datetime
 from typing import Any, Callable
 
-from pydantic import ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 from extensions.nodes.torch_config import TorchConfig
 from runnable import PythonJob, datastore, defaults
@@ -120,7 +120,8 @@ class TorchNode(DistributedNode, TorchConfig):
         return cls(executable=executable, **node_config, **task_config)
 
     def get_launch_config(self) -> LaunchConfig:
-        config, _, _ = config_from_args(self)
+        easy_model = EasyModel(**self.model_dump(exclude_none=True))
+        config, _, _ = config_from_args(easy_model)
         config.run_id = self._context.run_id
         return config
 
@@ -194,3 +195,9 @@ class TorchNode(DistributedNode, TorchConfig):
         assert (
             map_variable is None or not map_variable
         ), "TorchNode does not support map_variable"
+
+
+class EasyModel(BaseModel):
+    model_config = ConfigDict(extra="allow")
+    logs_specs: str | None = None
+    local_addr: str | None = Field(default=None)

runnable-0.30.5/extensions/nodes/torch_config.py

@@ -0,0 +1,39 @@
+from enum import Enum
+
+from pydantic import BaseModel, Field
+
+
+class StartMethod(str, Enum):
+    spawn = "spawn"
+    fork = "fork"
+    forkserver = "forkserver"
+
+
+class TorchConfig(BaseModel):
+    nnodes: str = Field(default="1:1")
+    nproc_per_node: int = Field(default=1)
+
+    rdzv_backend: str | None = Field(default="static")
+    rdzv_endpoint: str | None = Field(default="")
+    rdzv_id: str | None = Field(default="none")
+    rdzv_conf: str | None = Field(default="")
+
+    max_restarts: int | None = Field(default=None)
+    monitor_interval: float | None = Field(default=0.1)
+    start_method: str | None = Field(default=StartMethod.spawn)
+    role: str | None = Field(default="default")
+    log_dir: str | None = Field(default="torch_logs")
+    redirects: str | None = Field(default="0")
+    tee: str | None = Field(default="0")
+    master_addr: str | None = Field(default="localhost")
+    master_port: str | None = Field(default="29500")
+    training_script: str = Field(default="dummy_training_script")
+    training_script_args: str = Field(default="")
+
+    # Optional fields
+    local_ranks_filter: str | None = Field(default="0")
+    node_rank: int | None = Field(default=0)
+    standalone: bool | None = Field(default=None)
+    module: bool | None = Field(default=False)
+    no_python: bool | None = Field(default=False)
+    run_path: bool | None = Field(default=False)

{runnable-0.30.4 → runnable-0.30.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "runnable"
-version = "0.30.4"
+version = "0.30.5"
 description = "Add your description here"
 readme = "README.md"
 authors = [

runnable-0.30.4/extensions/nodes/torch_config.py

@@ -1,33 +0,0 @@
-from pydantic import BaseModel, Field
-
-
-class TorchConfig(BaseModel):
-    nnodes: str = Field(default="1:1")
-    nproc_per_node: int = Field(default=4)
-
-    rdzv_backend: str = Field(default="static")
-    rdzv_endpoint: str = Field(default="")
-    rdzv_id: str | None = Field(default=None)
-    rdzv_conf: str = Field(default="")
-
-    max_restarts: int = Field(default=3)
-    monitor_interval: float = Field(default=0.1)
-    start_method: str = Field(default="spawn")
-    role: str = Field(default="default_role")
-    log_dir: str = Field(default="torch_logs")
-    redirects: str = Field(default="1")
-    tee: str = Field(default="1")
-    master_addr: str = Field(default="localhost")
-    master_port: str = Field(default="29500")
-    training_script: str = Field(default="dummy_training_script")
-    training_script_args: str = Field(default="")
-
-    # Optional fields
-    local_ranks_filter: str = Field(default="")
-    node_rank: int = Field(default=0)
-    local_addr: str | None = Field(default=None)
-    logs_specs: str | None = Field(default=None)
-    standalone: bool = Field(default=False)
-    module: bool = Field(default=False)
-    no_python: bool = Field(default=False)
-    run_path: bool = Field(default=False)