save state

Signed-off-by: Ayush Kamat <[email protected]>
latchbio · Dec 23, 2024 · b471a43 · b471a43
1 parent ce4421a
commit b471a43
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 12 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -12,7 +12,7 @@ include = ["src/**/*.py", "src/latch_cli/services/init/*"]
 
 [project]
 name = "latch"
-version = "2.54.0.a9"
+version = "2.55.0.a6"
 description = "The Latch SDK"
 authors = [{ name = "Kenny Workman", email = "[email protected]" }]
 maintainers = [
@@ -34,7 +34,7 @@ dependencies = [
   "scp>=0.14.0",
   "boto3>=1.26.0",
   "tqdm>=4.63.0",
-  "lytekit==0.15.15",
+  "lytekit==0.15.17",
   "lytekitplugins-pods==0.6.2",
   "typing-extensions>=4.12.0",
   "apscheduler>=3.10.0",

diff --git a/src/latch_cli/docker_utils/__init__.py b/src/latch_cli/docker_utils/__init__.py
@@ -48,11 +48,6 @@ class DockerfileBuilder:
     direnv: Optional[Path] = None
 
     def get_prologue(self):
-        if self.wf_type == WorkflowType.snakemake:
-            library_name = '"latch[snakemake]"'
-        else:
-            library_name = "latch"
-
         self.commands.append(
             DockerCmdBlock(
                 comment="Prologue",
@@ -79,21 +74,25 @@ def get_prologue(self):
                     "env LANG='en_US.UTF-8'",
                     "",
                     "arg DEBIAN_FRONTEND=noninteractive",
-                    "",
-                    "# Latch SDK",
-                    "# DO NOT REMOVE",
-                    f"run pip install {library_name}=={self.config.latch_version}",
-                    "run mkdir /opt/latch",
                 ],
                 order=DockerCmdBlockOrder.precopy,
             )
         )
 
     def get_epilogue(self):
+        if self.wf_type == WorkflowType.snakemake:
+            library_name = '"latch[snakemake]"'
+        else:
+            library_name = "latch"
+
         self.commands.append(
             DockerCmdBlock(
                 comment="Epilogue",
                 commands=[
+                    "",
+                    "# Latch SDK",
+                    "# DO NOT REMOVE",
+                    f"run pip install {library_name}=={self.config.latch_version}",
                     "",
                     "# Latch workflow registration metadata",
                     "# DO NOT CHANGE",

diff --git a/src/latch_cli/snakemake/v2/workflow.py b/src/latch_cli/snakemake/v2/workflow.py
@@ -94,6 +94,8 @@ def snakemake_runtime(pvc_name: str, {parameters}):
         str(config_path),
         "--executor",
         "latch",
+        "--default-storage-provider",
+        "latch",
         "--jobs",
         "1000",
     ]