nshtrainer 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

nshtrainer/nn/nonlinearity.py

@@ -1,14 +1,13 @@
 from abc import ABC, abstractmethod
 from typing import Annotated, Literal
 
+import nshconfig as C
 import torch
 import torch.nn as nn
 from typing_extensions import override
 
-from ..config import Field, TypedConfig
 
-
-class BaseNonlinearityConfig(TypedConfig, ABC):
+class BaseNonlinearityConfig(C.Config, ABC):
     @abstractmethod
     def create_module(self) -> nn.Module:
         pass
@@ -153,5 +152,5 @@ NonlinearityConfig = Annotated[
     | SiLUNonlinearityConfig
     | MishNonlinearityConfig
     | SwiGLUNonlinearityConfig,
-    Field(discriminator="name"),
+    C.Field(discriminator="name"),
 ]

nshtrainer/optimizer.py

@@ -2,14 +2,13 @@ from abc import ABC, abstractmethod
 from collections.abc import Iterable
 from typing import Annotated, Any, Literal, TypeAlias
 
+import nshconfig as C
 import torch.nn as nn
 from torch.optim import Optimizer
 from typing_extensions import override
 
-from .config import Field, TypedConfig
 
-
-class OptimizerConfigBase(TypedConfig, ABC):
+class OptimizerConfigBase(C.Config, ABC):
     @abstractmethod
     def create_optimizer(
         self,
@@ -56,7 +55,4 @@ class AdamWConfig(OptimizerConfigBase):
         )
 
 
-OptimizerConfig: TypeAlias = Annotated[
-    AdamWConfig,
-    Field(discriminator="name"),
-]
+OptimizerConfig: TypeAlias = Annotated[AdamWConfig, C.Field(discriminator="name")]

nshtrainer/runner.py

@@ -1,8 +1,8 @@
-from dataclasses import dataclass
 from typing import Generic
 
+from nshrunner import RunInfo
 from nshrunner import Runner as _Runner
-from typing_extensions import Concatenate, TypeVar, TypeVarTuple, Unpack, override
+from typing_extensions import TypeVar, TypeVarTuple, Unpack, override
 
 from .model.config import BaseConfig
 
@@ -11,11 +11,21 @@ TArguments = TypeVarTuple("TArguments")
 TReturn = TypeVar("TReturn", infer_variance=True)
 
 
-@dataclass(frozen=True)
 class Runner(
     _Runner[Unpack[tuple[TConfig, Unpack[TArguments]]], TReturn],
     Generic[TConfig, Unpack[TArguments], TReturn],
 ):
     @override
-    def default_validate_fn():
-        pass
+    @classmethod
+    def default_validate_fn(cls, config: TConfig, *args: Unpack[TArguments]) -> None:
+        super().default_validate_fn(config, *args)
+
+    @override
+    @classmethod
+    def default_info_fn(cls, config: TConfig, *args: Unpack[TArguments]) -> RunInfo:
+        run_info = super().default_info_fn(config, *args)
+        return {
+            **run_info,
+            "id": config.id,
+            "base_dir": config.directory.project_root,
+        }

nshtrainer/trainer/signal_connector.py

@@ -25,14 +25,21 @@ _SIGNUM = int | signal.Signals
 _HANDLER: TypeAlias = Callable[[_SIGNUM, FrameType], Any] | int | signal.Handlers | None
 
 
-class _SignalConnector(_LightningSignalConnector):
-    def _auto_requeue_signals(self) -> list[signal.Signals]:
-        from ..model.base import BaseConfig
+def _resolve_requeue_signals():
+    signals: list[signal.Signals] = []
+
+    if timeout_signal_name := os.environ.get("NSHRUNNER_TIMEOUT_SIGNAL"):
+        signals.append(signal.Signals[timeout_signal_name])
 
-        if not isinstance(config := self.trainer.lightning_module.hparams, BaseConfig):
-            return []
+    if preempt_signal_name := os.environ.get("NSHRUNNER_PREEMPT_SIGNAL"):
+        signals.append(signal.Signals[preempt_signal_name])
 
-        signals = config.runner.submit._resolved_auto_requeue_signals()
+    return signals
+
+
+class _SignalConnector(_LightningSignalConnector):
+    def _auto_requeue_signals(self) -> list[signal.Signals]:
+        signals = _resolve_requeue_signals()
         signals_set = set(signals)
         valid_signals: set[signal.Signals] = signal.valid_signals()
         assert signals_set.issubset(
@@ -42,25 +49,29 @@ class _SignalConnector(_LightningSignalConnector):
 
     def _compose_and_register(
         self,
-        signum: _SIGNUM,
+        signum: signal.Signals,
         handlers: list[_HANDLER],
         replace_existing: bool = False,
     ):
         if self._is_on_windows():
-            log.info(f"Signal {signum} has no handlers or is not supported on Windows.")
+            log.info(
+                f"Signal {signum.name} has no handlers or is not supported on Windows."
+            )
             return
 
         if self._has_already_handler(signum):
             if not replace_existing:
                 log.info(
-                    f"Signal {signum} already has a handler. Adding ours to the existing one."
+                    f"Signal {signum.name} already has a handler. Adding ours to the existing one."
                 )
                 handlers.append(signal.getsignal(signum))
             else:
-                log.info(f"Replacing existing handler for signal {signum} with ours.")
+                log.info(
+                    f"Replacing existing handler for signal {signum.name} with ours."
+                )
 
         self._register_signal(signum, _HandlersCompose(handlers))
-        log.info(f"Registered {len(handlers)} handlers for signal {signum}.")
+        log.info(f"Registered {len(handlers)} handlers for signal {signum.name}.")
 
     @override
     def register_signal_handlers(self) -> None:

nshtrainer/trainer/trainer.py

@@ -31,7 +31,7 @@ log = logging.getLogger(__name__)
 
 def _is_bf16_supported_no_emulation():
     r"""Return a bool indicating if the current CUDA/ROCm device supports dtype bfloat16."""
-    version = cast(Any, torch.version)
+    version = getattr(torch, "version")
 
     # Check for ROCm, if true return true, no ROCM_VERSION check required,
     # since it is supported on AMD GPU archs.

nshtrainer/typecheck.py

@@ -82,6 +82,7 @@ def typecheck_this_module(additional_modules: Sequence[str] = ()):
     frame = get_frame(1)
     assert frame is not None, "frame is None"
     calling_module_name = get_frame_package_name(frame)
+    assert calling_module_name is not None, "calling_module_name is None"
 
     # Typecheck the calling module + any additional modules.
     typecheck_modules((calling_module_name, *additional_modules))

{nshtrainer-0.1.0.dist-info → nshtrainer-0.1.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: nshtrainer
-Version: 0.1.0
+Version: 0.1.1
 Summary: 
 Author: Nima Shoghi
 Author-email: nimashoghi@gmail.com
@@ -9,10 +9,21 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: beartype (>=0.18.5,<0.19.0)
+Requires-Dist: jaxtyping (>=0.2.33,<0.3.0)
+Requires-Dist: lightning
+Requires-Dist: lovely-numpy (>=0.2.13,<0.3.0)
+Requires-Dist: lovely-tensors (>=0.1.16,<0.2.0)
 Requires-Dist: nshconfig (>=0.2.0,<0.3.0)
-Requires-Dist: nshrunner (>=0.1.0,<0.2.0)
+Requires-Dist: nshrunner (>=0.5.3,<0.6.0)
+Requires-Dist: numpy
+Requires-Dist: pysnooper
+Requires-Dist: pytorch-lightning
+Requires-Dist: rich
 Requires-Dist: torch
+Requires-Dist: torchmetrics
 Requires-Dist: typing-extensions
+Requires-Dist: wrapt
 Description-Content-Type: text/markdown
 
 

{nshtrainer-0.1.0.dist-info → nshtrainer-0.1.1.dist-info}/RECORD

@@ -1,22 +1,16 @@
-nshtrainer/__init__.py,sha256=o39TbnjwUYzE4POcncUiDx02Ey-Hzx8UGuwJDjMcKZU,2971
+nshtrainer/__init__.py,sha256=OHbxLxVvFGW--ecuIGqkoylSVHFS4x4F1-oeuENH-Do,2212
 nshtrainer/_experimental/__init__.py,sha256=2tQIcrWT8U8no_AeBTYnozaTmxN40kuAJdGQ4b-PoWM,120
 nshtrainer/_experimental/flops/__init__.py,sha256=edo9Ez3LlrnxkNRX9W6YBhPkRPKYGLpkpnl5gx7sEX8,1550
 nshtrainer/_experimental/flops/flop_counter.py,sha256=-sL0Fy6poXa__hyzUMdZScjPULp4coQELQpPU6p6dXU,25736
 nshtrainer/_experimental/flops/module_tracker.py,sha256=bUL-IRTd0aF_DwmXkZjHZAA31p4ZEhyqhc26XWKQUUY,4922
 nshtrainer/_snoop.py,sha256=Rofv1Rd92E0LY40G3A-o9Hu0ZI73RR59wJD5l4Q3PDM,7022
-nshtrainer/_submit/print_environment_info.py,sha256=enbJGl_iHIlhKN8avzKnoZSb0zUQ_fUdnsQ8a_9tbYk,963
-nshtrainer/_submit/session/_output.py,sha256=CNGH5W6_XxAC5-TRvMAMxOHd3fjGpJhK-7RGTDyvMu4,245
-nshtrainer/_submit/session/_script.py,sha256=0AeBgBduDsoIEBrY9kebARiBUEGc50JAD9oE_IDiLnA,3775
-nshtrainer/_submit/session/lsf.py,sha256=p19EP6OhROZxqfRhzeTD7GDmfYaREIKMXMOI8G933FE,14307
-nshtrainer/_submit/session/slurm.py,sha256=JpAjQvck4LjGN8o8fOvIeMuFqrg1cioANoVsX5hU-3g,17594
-nshtrainer/_submit/session/unified.py,sha256=gfh-AtnMyFHzcQOUlhlAR__vaWDk1r9XCivz_t_lHKk,11695
 nshtrainer/actsave/__init__.py,sha256=G1T-fELuGWkVqdhdyoePtj2dTOUtcIOW4VgsXv9JNTA,338
 nshtrainer/actsave/_callback.py,sha256=QoTa60F70f1RxB41VKixN9l5_htfFQxXDPHHSNFreuk,2770
 nshtrainer/actsave/_loader.py,sha256=fAhD32DrJa4onkYfcwc21YIeGEYzOSXCK_HVo9SZLgQ,4604
 nshtrainer/actsave/_saver.py,sha256=0EHmQDhqVxQWRWWSyt03eP1K9ETiACMQYmsZkDMt6HY,9451
-nshtrainer/callbacks/__init__.py,sha256=ohE_MO_kX1o4SZwcipIXUA9m7XYcijEKJtGcoU8dTkY,1667
+nshtrainer/callbacks/__init__.py,sha256=I6W33ityL9Ko8jjqHh3WH_8miV59SAe9LxInhoqX5XE,1665
 nshtrainer/callbacks/_throughput_monitor_callback.py,sha256=aJo_11rc4lo0IYOd-kHmPDtzdC4ctgXyRudkRJqH4m4,23184
-nshtrainer/callbacks/base.py,sha256=WESZz1VSTl1xSGVXBmxFqWwbLxXcJp97jpg9zrE0EsY,3560
+nshtrainer/callbacks/base.py,sha256=LrcRUV02bZEKXRIRvhHT9qsvw_kwoWiAdQkVMyKc5NU,3542
 nshtrainer/callbacks/early_stopping.py,sha256=jriSU761wf_qTJ9Bos0D3h5aDvZHYpRqK62Ne8aWp5I,3768
 nshtrainer/callbacks/ema.py,sha256=zKCtvzZFo0ORlwNZHjaMk-sJoxrlTtFWOzR-yGy95W0,12134
 nshtrainer/callbacks/finite_checks.py,sha256=kX3TIJsxyqx0GuLJfYsqVgKU27zwjG9Z8324lyCFtwM,2087
@@ -30,17 +24,16 @@ nshtrainer/callbacks/print_table.py,sha256=FcA-CBWwMf9c1NNRinvYpZC400RNQxuP28bJf
 nshtrainer/callbacks/throughput_monitor.py,sha256=YQLdpX3LGybIiD814yT9yCCVSEXRWf8WwsvVaN5aDBE,1848
 nshtrainer/callbacks/timer.py,sha256=sDXPPcdDKu5xnuK_bjr8plIq9MBuluNJ42Mt9LvPZzc,4610
 nshtrainer/callbacks/wandb_watch.py,sha256=pUpMsNxd03ex1rzOmFw2HzGOXjnQGaH84m8cc2dXo4g,2937
-nshtrainer/config.py,sha256=0Fj5w-ry0BRl2_zJI6jwCnmMWE3p_eD8_Wn-NyFkTqU,10442
 nshtrainer/data/__init__.py,sha256=7mk1tr7SWUZ7ySbsf0y0ZPszk7u4QznPhQ-7wnpH9ec,149
 nshtrainer/data/balanced_batch_sampler.py,sha256=bcJBcQjh1hB1yKF_xSlT9AtEWv0BJjYc1CuH2BF-ea8,4392
 nshtrainer/data/transform.py,sha256=JeGxvytQly8hougrsdMmKG8gJ6qvFPDglJCO4Tp6STk,1795
-nshtrainer/lr_scheduler/__init__.py,sha256=GNGmkcJD3jgCMk7pfaanAYrKz9957qkx6_Q0rssiHK0,738
-nshtrainer/lr_scheduler/_base.py,sha256=1tWMABevKZAuGhJN8Me2E9eqEyqoLtsG0bADPjED7a4,3752
-nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=VhsxZJ_Mw9zjkAGunFQ1KRub5_QM5NRqaEFWtmedFp8,5212
-nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=Ct-uLo8Q4t7lJ_HwoLRhNmudnCw4cSnblpBEg22aVTI,2691
-nshtrainer/model/__init__.py,sha256=PdvZkpAVkqvCLipGJvEHFU3WxnSMxYpvtuOkvLIenxg,2078
-nshtrainer/model/base.py,sha256=bhngGHxr0suQB9Ezi_3d5JgDWYqS_yPgGJZrGmc1TnI,23571
-nshtrainer/model/config.py,sha256=RMDdrbtvwm5vTFPxQ2x1hqiBIEEE-OAknhF6KTWfkkk,70293
+nshtrainer/lr_scheduler/__init__.py,sha256=uEvgaFAs-4s_bAEMaildy0GT6OvgpgOEKTuzqutESHE,736
+nshtrainer/lr_scheduler/_base.py,sha256=7xOIuxQ86YHbFWG5a3gX46emQj1WN_LaY4-i0Q1TDBg,3659
+nshtrainer/lr_scheduler/linear_warmup_cosine.py,sha256=mn6cyizyI_stkXtg6zxIEGF9btIxMRWigUHUTlUYCSw,5221
+nshtrainer/lr_scheduler/reduce_lr_on_plateau.py,sha256=2ZdlV0RUMwg2DClzqYHr8_EKT1jZBUlSD39e-XlCsC4,2764
+nshtrainer/model/__init__.py,sha256=y32Hla-5whpzLL2BtCJpBakSp8o-1nQbpO0j_-xq_Po,1864
+nshtrainer/model/base.py,sha256=EMkOtp4YWGPHM0HPSTLbx75T9vlYmXO4XyD725xU70w,21453
+nshtrainer/model/config.py,sha256=6lATW6-Z1SIDgQ1IWrGBVQKTr8DhL5b_rFbJHQz0d5o,66796
 nshtrainer/model/modules/callback.py,sha256=JF59U9-CjJsAIspEhTJbVaGN0wGctZG7UquE3IS7R8A,6408
 nshtrainer/model/modules/debug.py,sha256=DTVty8cKnzj1GCULRyGx_sWTTsq9NLi30dzqjRTnuCU,1127
 nshtrainer/model/modules/distributed.py,sha256=ABpR9d-3uBS_fivfy_WYW-dExW6vp5BPaoPQnOudHng,1725
@@ -52,21 +45,20 @@ nshtrainer/nn/__init__.py,sha256=57LPaP3G-BBGD2eGxbBUABNgYl3s_oASwrtOSS4bzTs,133
 nshtrainer/nn/mlp.py,sha256=i-dHk0tomO_XlU6cKN4CC4HxTaYb-ukBCAgY1ySXl4I,3963
 nshtrainer/nn/module_dict.py,sha256=NOY0B6WDTnktyWH4GthsprMQo0bpehC-hCq9SfD8paE,2329
 nshtrainer/nn/module_list.py,sha256=fb2u5Rqdjff8Pekyr9hkCPkBorQ-fldzzFAjsgWAm30,1719
-nshtrainer/nn/nonlinearity.py,sha256=IhIR8NCTY3Np9dMDnUouERR8ZhWpK3S0hTbT0i8HezU,3645
-nshtrainer/optimizer.py,sha256=JiLNRtcfYxyhAab1Z1QcEzmrX9S_JyrBS67TXy12kXI,1557
-nshtrainer/runner.py,sha256=9HsYB58aasY9RVvya_gPECDs_MBhM1fl4cbM3iJYTDc,600
+nshtrainer/nn/nonlinearity.py,sha256=owtU4kh4G98psD0axOJWVfBhm-OtJVgFM-TXSHmbNPU,3625
+nshtrainer/optimizer.py,sha256=kuJEA1pvB3y1FcsfhAoOJujVqEZqFHlmYO8GW6JeA1g,1527
+nshtrainer/runner.py,sha256=af_EGnQTSvUgwnVhhytvY3V7o_Xg-xx-sLb8K2Szb1E,979
 nshtrainer/scripts/check_env.py,sha256=IMl6dSqsLYppI0XuCsVq8lK4bYqXwY9KHJkzsShz4Kg,806
 nshtrainer/scripts/find_packages.py,sha256=FbdlfmAefttFSMfaT0A46a-oHLP_ioaQKihwBfBeWeA,1467
 nshtrainer/trainer/__init__.py,sha256=P2rmr8oBVTHk-HJHYPcUwWqDEArMbPR4_rPpATbWK3E,40
-nshtrainer/trainer/signal_connector.py,sha256=aGg6kRiHiqtAdGlEvEvGLmOy7AvRHTSkXdTmZpRXbjU,8435
-nshtrainer/trainer/trainer.py,sha256=oi8KdHF1AdZ54KFbCFAEI7W-C7qRtRe-KtOjNwBuS3M,14033
-nshtrainer/typecheck.py,sha256=CFkmPIxCU24CHk_7_pykb-Y1PRNhpLgsVZw1zuuOS_U,4614
+nshtrainer/trainer/signal_connector.py,sha256=QAoPM_C5JJOVQebcrJOimUUD3GHyoeZUqCEAvzZlT4U,8710
+nshtrainer/trainer/trainer.py,sha256=eYEYfY9v70MuorHcSf8nqM7f2CkmUHhpPcjCk4FJD7k,14034
+nshtrainer/typecheck.py,sha256=RGYHxDBcs97E6ayl6Olc43JBZXQolCtMxcLBniVCVBg,4688
 nshtrainer/util/environment.py,sha256=_SEtiQ_s5bL5pllUlf96AOUv15kNvCPvocVC13S7mIk,4166
 nshtrainer/util/seed.py,sha256=HEXgVs-wldByahOysKwq7506OHxdYTEgmP-tDQVAEkQ,287
-nshtrainer/util/singleton.py,sha256=nLhpuMZxl0zdNsnvS97o4ASUnKzCWYEKLzR_j9oP_xs,2208
 nshtrainer/util/slurm.py,sha256=rofIU26z3SdL79SF45tNez6juou1cyDLz07oXEZb9Hg,1566
 nshtrainer/util/typed.py,sha256=NGuDkDzFlc1fAoaXjOFZVbmj0mRFjsQi1E_hPa7Bn5U,128
 nshtrainer/util/typing_utils.py,sha256=8ptjSSLZxlmy4FY6lzzkoGoF5fGNClo8-B_c0XHQaNU,385
-nshtrainer-0.1.0.dist-info/METADATA,sha256=3zdNPxyB-I6Gudq2gTaU0crdgmDCcGCp6Zudef0DtuM,529
-nshtrainer-0.1.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-nshtrainer-0.1.0.dist-info/RECORD,,
+nshtrainer-0.1.1.dist-info/METADATA,sha256=32iVLvdJh6OJQyD-_7NDO6IYqfHPSflDznYfYaCo8-c,882
+nshtrainer-0.1.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+nshtrainer-0.1.1.dist-info/RECORD,,

nshtrainer/_submit/print_environment_info.py

@@ -1,31 +0,0 @@
-import logging
-import os
-import sys
-
-
-def print_environment_info(log: logging.Logger | None = None):
-    if log is None:
-        logging.basicConfig(level=logging.INFO)
-        log = logging.getLogger(__name__)
-
-    log_message_lines: list[str] = []
-    log_message_lines.append("Python executable: " + sys.executable)
-    log_message_lines.append("Python version: " + sys.version)
-    log_message_lines.append("Python prefix: " + sys.prefix)
-    log_message_lines.append("Python path:")
-    for path in sys.path:
-        log_message_lines.append(f"  {path}")
-
-    log_message_lines.append("Environment variables:")
-    for key, value in os.environ.items():
-        log_message_lines.append(f"  {key}={value}")
-
-    log_message_lines.append("Command line arguments:")
-    for i, arg in enumerate(sys.argv):
-        log_message_lines.append(f"  {i}: {arg}")
-
-    log.critical("\n".join(log_message_lines))
-
-
-if __name__ == "__main__":
-    print_environment_info()

nshtrainer/_submit/session/_output.py

@@ -1,12 +0,0 @@
-from dataclasses import dataclass
-from pathlib import Path
-
-
-@dataclass(frozen=True)
-class SubmitOutput:
-    command_parts: list[str]
-    script_path: Path
-
-    @property
-    def command(self) -> str:
-        return " ".join(self.command_parts)

nshtrainer/_submit/session/_script.py

@@ -1,109 +0,0 @@
-from collections.abc import Iterable, Mapping, Sequence
-from pathlib import Path
-
-
-def _create_launcher_script_file(
-    script_path: Path,
-    original_command: str | Iterable[str],
-    environment: Mapping[str, str],
-    setup_commands: Sequence[str],
-    chmod: bool = True,
-    prepend_command_with_exec: bool = True,
-    command_prefix: str | None = None,
-    # ^ If True, the original command will be prepended with 'exec' to replace the shell process
-    #   with the command. This is useful for ensuring that the command is the only process in the
-    #   process tree (e.g. for better signal handling).
-):
-    """
-    Creates a helper bash script for running the given function.
-
-    The core idea: The helper script is essentially one additional layer of indirection
-    that allows us to encapsulates the environment setup and the actual function call
-    in a single bash script (that does not require properly set up Python environment).
-
-    In effect, this allows us to, for example:
-    - Easily run the function in the correct environment
-        (without having to deal with shell hooks)
-        using `conda run -n myenv bash /path/to/helper.sh`.
-    - Easily run the function in a Singularity container
-        using `singularity exec my_container.sif bash /path/to/helper.sh`.
-    """
-    with script_path.open("w") as f:
-        f.write("#!/bin/bash\n\n")
-        f.write("set -e\n\n")
-
-        if environment:
-            for key, value in environment.items():
-                f.write(f"export {key}={value}\n")
-            f.write("\n")
-
-        if setup_commands:
-            for setup_command in setup_commands:
-                f.write(f"{setup_command}\n")
-            f.write("\n")
-
-        if not isinstance(original_command, str):
-            original_command = " ".join(original_command)
-
-        if command_prefix:
-            original_command = f"{command_prefix} {original_command}"
-
-        if prepend_command_with_exec:
-            original_command = f"exec {original_command}"
-        f.write(f"{original_command}\n")
-
-    if chmod:
-        # Make the script executable
-        script_path.chmod(0o755)
-
-
-def write_helper_script(
-    base_dir: Path,
-    command: str | Iterable[str],
-    environment: Mapping[str, str],
-    setup_commands: Sequence[str],
-    chmod: bool = True,
-    prepend_command_with_exec: bool = True,
-    command_prefix: str | None = None,
-    file_name: str = "helper.sh",
-):
-    """
-    Creates a helper bash script for running the given function.
-
-    The core idea: The helper script is essentially one additional layer of indirection
-    that allows us to encapsulates the environment setup and the actual function call
-    in a single bash script (that does not require properly set up Python environment).
-
-    In effect, this allows us to, for example:
-    - Easily run the function in the correct environment
-        (without having to deal with shell hooks)
-        using `conda run -n myenv bash /path/to/helper.sh`.
-    - Easily run the function in a Singularity container
-        using `singularity exec my_container.sif bash /path/to/helper.sh`.
-    """
-
-    out_path = base_dir / file_name
-    _create_launcher_script_file(
-        out_path,
-        command,
-        environment,
-        setup_commands,
-        chmod,
-        prepend_command_with_exec,
-        command_prefix,
-    )
-    return out_path
-
-
-DEFAULT_TEMPLATE = "bash {script}"
-
-
-def helper_script_to_command(script: Path, template: str | None) -> str:
-    if not template:
-        template = DEFAULT_TEMPLATE
-
-    # Make sure the template has '{script}' in it
-    if "{script}" not in template:
-        raise ValueError(f"Template must contain '{{script}}'. Got: {template!r}")
-
-    return template.format(script=str(script.absolute()))