latch 2.76.6__tar.gz → 2.76.7__tar.gz

{latch-2.76.6 → latch-2.76.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: latch
-Version: 2.76.6
+Version: 2.76.7
 Summary: The Latch SDK
 Project-URL: Homepage, https://latch.bio
 Project-URL: Documentation, https://latch.wiki

{latch-2.76.6 → latch-2.76.7}/pyproject.toml

@@ -12,7 +12,7 @@ include = ["src/**/*.py", "src/**/*.graphql", "src/**/py.typed", "src/latch_cli/
 
 [project]
 name = "latch"
-version = "2.76.6"
+version = "2.76.7"
 description = "The Latch SDK"
 authors = [{ name = "Kenny Workman", email = "kenny@latch.bio" }]
 maintainers = [{ name = "Kenny Workman", email = "kenny@latch.bio" }]
@@ -247,6 +247,10 @@ ignore = [
   "PLC1901",
 ]
 
+[tool.ruff.lint.pyupgrade]
+# Pydantic resolves annotations at runtime, where PEP 604 is unsupported on Python 3.9.
+keep-runtime-typing = true
+
 [tool.ruff.format]
 preview = true
 skip-magic-trailing-comma = true

{latch-2.76.6 → latch-2.76.7}/src/latch/resources/launch_plan.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import base64
-from typing import Any
+from typing import Any, Optional
 
 from flytekit.core.launch_plan import LaunchPlan as _LaunchPlan
 from flytekit.core.workflow import PythonFunctionWorkflow
@@ -41,7 +41,7 @@ class LaunchPlan:
         name: str,
         default_params: dict[str, Any],
         *,
-        description: str | None = None,
+        description: Optional[str] = None,
     ):
         labels = {}
         if description is not None:

{latch-2.76.6 → latch-2.76.7}/src/latch_cli/services/k8s/develop.py

@@ -306,7 +306,7 @@ def local_development(
     *,
     size: InstanceSize,
     skip_confirm_dialog: bool,
-    wf_version: str | None,
+    wf_version: Optional[str],
     disable_sync: bool,
 ):
     if not disable_sync:

{latch-2.76.6 → latch-2.76.7}/src/latch_cli/services/launch/type_converter.py

@@ -2,7 +2,7 @@ import dataclasses
 from dataclasses import asdict, is_dataclass
 from datetime import datetime, timedelta
 from enum import Enum
-from typing import Any
+from typing import Any, Optional
 
 from flytekit.models import literals as _literals
 from flytekit.models import types as _types
@@ -59,7 +59,7 @@ def _convert_primitive(
     if simple_type == _types.SimpleType.STRUCT:  # pyright: ignore[reportUnnecessaryComparison]
         return _convert_to_struct(value)
 
-    primitive: _literals.Primitive | None = None
+    primitive: Optional[_literals.Primitive] = None
 
     if simple_type == _types.SimpleType.INTEGER:  # pyright: ignore[reportUnnecessaryComparison]
         if isinstance(value, Enum):
@@ -243,7 +243,7 @@ def _convert_blob(
     blob_type: _core_types.BlobType,
 ) -> _literals.Literal:
 
-    remote_uri: str | None = None
+    remote_uri: Optional[str] = None
     if isinstance(value, (LatchFile, LatchDir)):
         remote_uri = value.remote_path if value.remote_path is not None else value.path
     elif isinstance(value, LPath):

{latch-2.76.6 → latch-2.76.7}/src/latch_cli/services/pods.py

@@ -5,7 +5,7 @@ import os
 import shlex
 import time
 from importlib import resources
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING, Any, Optional
 from urllib.parse import urljoin
 
 import click
@@ -46,20 +46,20 @@ pod_ssh_starting_statuses = {
 class CreatePodRequest(BaseModel):
     model_config = ConfigDict(extra="forbid")
 
-    ws_account_id: str | None = Field(
+    ws_account_id: Optional[str] = Field(
         default=None, json_schema_extra={"default_description": "current workspace"}
     )
     display_name: str = Field(..., json_schema_extra={"template": "My Pod"})
     cpu: int = Field(..., json_schema_extra={"template": 2, "units": "cores"})
     memory: int = Field(..., json_schema_extra={"template": 8, "units": "GiB"})
     gpu: int = Field(default=0, json_schema_extra={"template": 0})
-    gpu_type: Literal["nvidia-a10g", "nvidia-l40s"] | None = Field(
+    gpu_type: Optional[Literal["nvidia-a10g", "nvidia-l40s"]] = Field(
         default=None, json_schema_extra={"template": None}
     )
     storage_gigs: int = Field(
         default=20, json_schema_extra={"template": 20, "units": "GiB"}
     )
-    backup_interval: Literal["daily", "weekly", "monthly"] | None = Field(
+    backup_interval: Optional[Literal["daily", "weekly", "monthly"]] = Field(
         default=None, json_schema_extra={"template": None}
     )
     target_region: Literal["us-west-2", "us-east-1", "eu-central-1", "eu-west-1"] = (
@@ -105,7 +105,7 @@ class PodInfosResponse(BaseModel):
 
 
 class PodSshResponse(BaseModel):
-    pod_info: dict[str, Any] | None = Field(alias="podInfo")
+    pod_info: Optional[dict[str, Any]] = Field(alias="podInfo")
 
 
 def pod_list_fields(*, detailed: bool) -> list[str]:
@@ -295,8 +295,8 @@ def _format_pod_status(status: str) -> str:
 
 
 def _get_pod_ssh_args(
-    pod_id: int, pod: dict[str, object], *, key: Path | None = None
-) -> list[str] | None:
+    pod_id: int, pod: dict[str, object], *, key: Optional[Path] = None
+) -> Optional[list[str]]:
     deployment = pod.get("deployment")
     if not isinstance(deployment, dict):
         click.secho("Pod deployment information is unavailable.", fg="red")
@@ -331,7 +331,7 @@ def ssh_pod(
     pod_id: int,
     *,
     print_only: bool = False,
-    key: Path | None = None,
+    key: Optional[Path] = None,
     poll_interval_seconds: float = pod_ssh_poll_interval_seconds,
     max_wait_seconds: float = pod_ssh_max_wait_seconds,
     exec_fn: Callable[[str, list[str]], object] = os.execvp,

{latch-2.76.6 → latch-2.76.7}/src/latch_cli/services/register/register.py

@@ -62,6 +62,17 @@ def _print_window(cur_lines: List[str], line: str):
 docker_build_step_pat = re.compile("^Step [0-9]+/[0-9]+ :")
 
 
+def _replace_task_protos(
+    protos: List[Path], new_protos: List[Path], task_name: str
+) -> List[Path]:
+    """Replace a task's default-image protobuf with its custom-image version."""
+    new_task_protos = {
+        proto.name: proto for proto in new_protos if task_name in proto.name
+    }
+
+    return [new_task_protos.get(proto.name, proto) for proto in protos]
+
+
 def print_and_write_build_logs(
     build_logs: Iterable[DockerBuildLogItem],
     image: str,
@@ -581,22 +592,7 @@ def register(
 
                     new_protos = _recursive_list(local_task_td)
 
-                    try:
-                        split_task_name = task_name.split(".")
-                        task_name = ".".join(
-                            split_task_name[split_task_name.index(ctx.wf_module) :]
-                        )
-                        for new_proto in new_protos:
-                            if task_name in new_proto.name:
-                                protos = [
-                                    new_proto if new_proto.name == f.name else f
-                                    for f in protos
-                                ]
-                    except ValueError as e:
-                        raise ValueError(
-                            f"Unable to match {task_name} to any of the protobuf files"
-                            f" in {new_protos}"
-                        ) from e
+                    protos = _replace_task_protos(protos, new_protos, task_name)
                 except TypeError as e:
                     raise ValueError(
                         "The path to your provided dockerfile ",

{latch-2.76.6 → latch-2.76.7}/src/latch_cli/services/workspace.py

@@ -19,7 +19,7 @@ def workspace(workspace_id: Optional[str] = None):
     """
     data = get_workspaces()
 
-    old_id: str | None
+    old_id: Optional[str]
     try:
         old_id = current_workspace()
     except NoWorkspaceSelectedError: