coiled 1.126.1.dev22__tar.gz → 1.126.1.dev24__tar.gz

{coiled-1.126.1.dev22 → coiled-1.126.1.dev24}/PKG-INFO

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

{coiled-1.126.1.dev22 → coiled-1.126.1.dev24}/coiled/cli/batch/run.py

@@ -196,6 +196,7 @@ def get_kwargs_from_header(f: dict, click_params: list):
         "default is to use the entrypoint (if any) set on the image."
     ),
 )
+@click.option("--run-on-host", default=None, help="Run code directly on host, not inside docker container.")
 @click.option(
     "--env",
     "-e",
@@ -773,7 +774,9 @@ def _batch_run(default_kwargs, logger=None, from_cli=False, **kwargs) -> dict:
         # if batch job is running in extra container, then we just need a pretty minimal dask container
         # so for now switch the default in that case to basic dask container
         # TODO would it be better to use a pre-built senv with our `cloud-env-run` container instead?
-        "container": dask_container if kwargs["container"] and not kwargs["software"] else None,
+        "container": dask_container
+        if (kwargs["container"] or kwargs.get("run_on_host")) and not kwargs["software"]
+        else None,
         "region": kwargs["region"],
         "scheduler_options": {
             "idle_timeout": "520 weeks",  # TODO allow job timeout?
@@ -829,6 +832,7 @@ def _batch_run(default_kwargs, logger=None, from_cli=False, **kwargs) -> dict:
         "pipe_to_files": bool(kwargs.get("pipe_to_files")),
         "host_setup": host_setup_content,
         "job_timeout_seconds": parse_timedelta(kwargs["job_timeout"]) if kwargs["job_timeout"] else None,
+        "run_in_container": not kwargs.get("run_on_host"),
     }
 
     with coiled.Cloud(workspace=kwargs["workspace"]) as cloud: