coiled 1.130.4.dev15__tar.gz → 1.130.5.dev1__tar.gz

{coiled-1.130.4.dev15 → coiled-1.130.5.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: coiled
-Version: 1.130.4.dev15
+Version: 1.130.5.dev1
 Summary: Python client for coiled.io dask clusters
 Project-URL: Homepage, https://coiled.io
 Maintainer-email: Coiled <info@coiled.io>

{coiled-1.130.4.dev15 → coiled-1.130.5.dev1}/coiled/cli/run.py

@@ -570,6 +570,12 @@ def get_entrypoint(connection, container_name) -> str:
     default=None,
     help="Non-default value for shm_size (for example, '3 GiB').",
 )
+@click.option(
+    "--skip-ssd-mount",
+    is_flag=True,
+    default=None,
+    help="Don't automatically mount SSD/NVMe (if any) to ``/scratch``.",
+)
 @click.option(
     "--filestore",
     "filestore_names",
@@ -607,6 +613,7 @@ def run(
     package_sync_strict,
     package_sync_conda_extras,
     docker_shm_size,
+    skip_ssd_mount,
     filestore_names,
     command,
 ):
@@ -644,6 +651,7 @@ def run(
         package_sync_conda_extras=package_sync_conda_extras,
         docker_shm_size=docker_shm_size,
         filestore_names=filestore_names,
+        skip_ssd_mount=skip_ssd_mount,
     )
     sys.exit(info["exit_code"])
 
@@ -682,6 +690,7 @@ def start_run(
     package_sync_conda_extras: List[str] | None = None,
     docker_shm_size: str | None = None,
     filestore_names: list[str] | None = None,
+    skip_ssd_mount: bool | None = None,
 ):
     runtime_env_dict = dict_from_key_val_list(env)
     tags = dict_from_key_val_list(tag)
@@ -778,6 +787,7 @@ def start_run(
                     "package_sync_conda_extras": package_sync_conda_extras,
                     "backend_options": {"docker_shm_size": docker_shm_size} if docker_shm_size else None,
                     "unset_single_threading_variables": True,
+                    "skip_ssd_mount": skip_ssd_mount,
                     "filestores_to_attach": filestores_to_attach,
                 }
                 cluster_kwargs["name"] = name or f"run-{short_random_string()}"

{coiled-1.130.4.dev15 → coiled-1.130.5.dev1}/coiled/v2/cluster.py

@@ -210,6 +210,7 @@ class ClusterKwargs(TypedDict, total=False):
     scheduler_sidecars: list[dict] | None
     worker_sidecars: list[dict] | None
     pause_on_exit: bool | None
+    skip_ssd_mount: bool | None
     filestores_to_attach: list[dict] | None
 
 
@@ -479,6 +480,8 @@ class Cluster(DistributedCluster, Generic[IsAsynchronous]):
         Like ``scheduler_sidecars``, but run on worker VMs instead of scheduler.
     pause_on_exit
         Pause the cluster instead of shutting it down when exiting.
+    skip_ssd_mount
+        Don't automatically mount SSD/NVMe (if any) to ``/scratch``.
     filestores_to_attach
         List of filestores to attach (specified as ``{"id": id, "input": True, "output": True}``, not name).
     """
@@ -566,6 +569,7 @@ class Cluster(DistributedCluster, Generic[IsAsynchronous]):
         scheduler_sidecars: list[dict] | None = None,
         worker_sidecars: list[dict] | None = None,
         pause_on_exit: bool | None = None,
+        skip_ssd_mount: bool | None = None,
         filestores_to_attach: list[dict] | None = None,
     ):
         self.pause_on_exit = pause_on_exit
@@ -866,6 +870,7 @@ class Cluster(DistributedCluster, Generic[IsAsynchronous]):
         self._asynchronous = asynchronous
         self._is_coiled_hosted = False
         self.shutdown_on_close = shutdown_on_close
+        self.skip_ssd_mount = skip_ssd_mount
 
         self.environ = normalize_environ(environ)
         aws_default_region = self._get_aws_default_region()
@@ -1622,6 +1627,7 @@ class Cluster(DistributedCluster, Generic[IsAsynchronous]):
                     worker_sidecars=self.worker_sidecars,
                     host_setup_script_content=self.host_setup_script_content,
                     pause_on_exit=self.pause_on_exit,
+                    skip_ssd_mount=self.skip_ssd_mount,
                     filestores_to_attach=self.filestores_to_attach,
                     cluster_timeout_seconds=self.cluster_timeout_seconds,
                 )

{coiled-1.130.4.dev15 → coiled-1.130.5.dev1}/coiled/v2/core.py

@@ -652,6 +652,7 @@ class CloudV2(OldCloud, Generic[IsAsynchronous]):
         worker_sidecars: list[dict] | None = None,
         host_setup_script_content: str | None = None,
         pause_on_exit: bool | None = None,
+        skip_ssd_mount: bool | None = None,
         cluster_timeout_seconds: int | None = None,
         filestores_to_attach: list[dict] | None = None,
     ) -> Tuple[int, bool]:
@@ -693,6 +694,7 @@ class CloudV2(OldCloud, Generic[IsAsynchronous]):
             "worker_sidecars": worker_sidecars,
             "host_setup_script": host_setup_script_content,
             "pause_on_exit": pause_on_exit,
+            "skip_ssd_mount": skip_ssd_mount,
             "cluster_timeout_seconds": cluster_timeout_seconds,
             "coiled_cloud_env_image": dask.config.get("coiled.cloud-env-image", None),
             "filestores_to_attach": filestores_to_attach,