latch 2.76.0__tar.gz → 2.76.2__tar.gz

{latch-2.76.0 → latch-2.76.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: latch
-Version: 2.76.0
+Version: 2.76.2
 Summary: The Latch SDK
 Project-URL: Homepage, https://latch.bio
 Project-URL: Documentation, https://latch.wiki
@@ -52,7 +52,7 @@ Requires-Dist: gql==3.5.0
 Requires-Dist: graphql-core==3.2.3
 Requires-Dist: kubernetes>=24.2.0
 Requires-Dist: latch-persistence>=0.1.5
-Requires-Dist: lytekit==0.15.35
+Requires-Dist: lytekit==0.16
 Requires-Dist: lytekitplugins-pods==0.7.4
 Requires-Dist: orjson>=3.10.12
 Requires-Dist: paramiko>=3.4.0

{latch-2.76.0 → latch-2.76.2}/pyproject.toml

@@ -12,7 +12,7 @@ include = ["src/**/*.py", "src/**/*.graphql", "src/**/py.typed", "src/latch_cli/
 
 [project]
 name = "latch"
-version = "2.76.0"
+version = "2.76.2"
 description = "The Latch SDK"
 authors = [{ name = "Kenny Workman", email = "kenny@latch.bio" }]
 maintainers = [{ name = "Kenny Workman", email = "kenny@latch.bio" }]
@@ -31,7 +31,7 @@ dependencies = [
   "scp>=0.14.0",
   "boto3>=1.26.0",
   "tqdm>=4.63.0",
-  "lytekit==0.15.35",
+  "lytekit==0.16",
   "lytekitplugins-pods==0.7.4",
   "typing-extensions>=4.12.0",
   "apscheduler>=3.10.0",

{latch-2.76.0 → latch-2.76.2}/src/latch_cli/centromere/ctx.py

@@ -94,6 +94,7 @@ class _CentromereCtx:
         use_new_centromere: bool = False,
         overwrite: bool = False,
         dockerfile_path: Optional[Path] = None,
+        account_id: Optional[str] = None,
     ):
         self.use_new_centromere = use_new_centromere
         self.remote = remote
@@ -114,7 +115,9 @@ class _CentromereCtx:
             log.debug("retrieve_or_login")
             self.token = retrieve_or_login()
             log.debug("current_workspace")
-            self.account_id = current_workspace()
+            self.account_id = (
+                account_id if account_id is not None else current_workspace()
+            )
 
             self.dkr_repo = config.dkr_repo
             log.debug("resolving pkg_root")
@@ -629,7 +632,7 @@ class _CentromereCtx:
 
         headers = {"Authorization": f"Bearer {self.token}"}
 
-        ws_id = current_workspace()
+        ws_id = self.account_id
         if ws_id is None or ws_id == "":
             ws_id = account_id_from_token(retrieve_or_login())
 

{latch-2.76.0 → latch-2.76.2}/src/latch_cli/services/register/register.py

@@ -374,6 +374,9 @@ def register(
             click.secho("\n`snakemake` package is not installed.", fg="red", bold=True)
             sys.exit(1)
 
+    if workspace_id is None:
+        workspace_id = current_workspace()
+
     with _CentromereCtx(
         Path(pkg_root),
         disable_auto_version=disable_auto_version,
@@ -385,6 +388,7 @@ def register(
         use_new_centromere=use_new_centromere,
         overwrite=skip_confirmation,
         dockerfile_path=dockerfile_path,
+        account_id=workspace_id,
     ) as ctx:
         assert ctx.workflow_name is not None, "Unable to determine workflow name"
         assert ctx.version is not None, "Unable to determine workflow version"
@@ -398,8 +402,6 @@ def register(
         )
         click.echo(" ".join([click.style("Version:", fg="bright_blue"), ctx.version]))
 
-        if workspace_id is None:
-            workspace_id = current_workspace()
         workspaces = get_workspaces()
 
         if workspace_id not in workspaces:

{latch-2.76.0 → latch-2.76.2}/src/latch_cli/services/register/utils.py

@@ -24,7 +24,6 @@ import click
 import docker
 import requests
 
-from latch.utils import current_workspace
 from latch_sdk_config.latch import config
 
 if TYPE_CHECKING:
@@ -41,7 +40,7 @@ def _docker_login(ctx: _CentromereCtx):
     assert ctx.dkr_client is not None
 
     headers = {"Authorization": f"Bearer {ctx.token}"}
-    data = {"pkg_name": ctx.image, "ws_account_id": current_workspace()}
+    data = {"pkg_name": ctx.image, "ws_account_id": ctx.account_id}
     response = requests.post(ctx.latch_image_api_url, headers=headers, json=data)
     log.debug("POST latch_image_api_url %s", response.status_code)