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

nshtrainer/__init__.py

@@ -10,16 +10,7 @@ from . import typecheck as typecheck
 from ._snoop import snoop as snoop
 from .actsave import ActLoad as ActLoad
 from .actsave import ActSave as ActSave
-from .config import MISSING as MISSING
-from .config import AllowMissing as AllowMissing
-from .config import Field as Field
-from .config import MissingField as MissingField
-from .config import PrivateAttr as PrivateAttr
-from .config import TypedConfig as TypedConfig
 from .data import dataset_transform as dataset_transform
-from .log import init_python_logging as init_python_logging
-from .log import lovely as lovely
-from .log import pretty as pretty
 from .lr_scheduler import LRSchedulerConfig as LRSchedulerConfig
 from .model import ActSaveConfig as ActSaveConfig
 from .model import Base as Base
@@ -41,24 +32,17 @@ from .model import (
     EnvironmentSLURMInformationConfig as EnvironmentSLURMInformationConfig,
 )
 from .model import GradientClippingConfig as GradientClippingConfig
-from .model import LightningDataModuleBase as LightningDataModuleBase
 from .model import LightningModuleBase as LightningModuleBase
 from .model import LoggingConfig as LoggingConfig
 from .model import MetricConfig as MetricConfig
 from .model import OptimizationConfig as OptimizationConfig
 from .model import PrimaryMetricConfig as PrimaryMetricConfig
-from .model import PythonLogging as PythonLogging
 from .model import ReproducibilityConfig as ReproducibilityConfig
-from .model import RunnerConfig as RunnerConfig
 from .model import SanityCheckingConfig as SanityCheckingConfig
-from .model import SeedConfig as SeedConfig
 from .model import TrainerConfig as TrainerConfig
 from .model import WandbWatchConfig as WandbWatchConfig
 from .nn import TypedModuleDict as TypedModuleDict
 from .nn import TypedModuleList as TypedModuleList
 from .optimizer import OptimizerConfig as OptimizerConfig
 from .runner import Runner as Runner
-from .runner import SnapshotConfig as SnapshotConfig
 from .trainer import Trainer as Trainer
-from .util.singleton import Registry as Registry
-from .util.singleton import Singleton as Singleton

nshtrainer/callbacks/__init__.py

@@ -1,6 +1,7 @@
 from typing import Annotated
 
-from ..config import Field
+import nshconfig as C
+
 from .base import CallbackConfigBase as CallbackConfigBase
 from .early_stopping import EarlyStopping as EarlyStopping
 from .ema import EMA as EMA
@@ -31,5 +32,5 @@ CallbackConfig = Annotated[
     | NormLoggingConfig
     | GradientSkippingConfig
     | EMAConfig,
-    Field(discriminator="name"),
+    C.Field(discriminator="name"),
 ]

nshtrainer/callbacks/base.py

@@ -4,10 +4,9 @@ from collections.abc import Iterable
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, TypeAlias, TypedDict
 
+import nshconfig as C
 from lightning.pytorch import Callback
 
-from ..config import TypedConfig
-
 if TYPE_CHECKING:
     from ..model.config import BaseConfig
 
@@ -20,7 +19,7 @@ class CallbackMetadataDict(TypedDict, total=False):
     """Priority of the callback. Callbacks with higher priority will be loaded first."""
 
 
-class CallbackMetadataConfig(TypedConfig):
+class CallbackMetadataConfig(C.Config):
     ignore_if_exists: bool = False
     """If `True`, the callback will not be added if another callback with the same class already exists."""
 
@@ -37,7 +36,7 @@ class CallbackWithMetadata:
 ConstructedCallback: TypeAlias = Callback | CallbackWithMetadata
 
 
-class CallbackConfigBase(TypedConfig, ABC):
+class CallbackConfigBase(C.Config, ABC):
     metadata: CallbackMetadataConfig = CallbackMetadataConfig()
     """Metadata for the callback."""
 

nshtrainer/lr_scheduler/__init__.py

@@ -1,6 +1,7 @@
 from typing import Annotated, TypeAlias
 
-from ..config import Field
+import nshconfig as C
+
 from ._base import LRSchedulerConfigBase as LRSchedulerConfigBase
 from ._base import LRSchedulerMetadata as LRSchedulerMetadata
 from .linear_warmup_cosine import (
@@ -14,5 +15,5 @@ from .reduce_lr_on_plateau import ReduceLROnPlateauConfig as ReduceLROnPlateauCo
 
 LRSchedulerConfig: TypeAlias = Annotated[
     LinearWarmupCosineDecayLRSchedulerConfig | ReduceLROnPlateauConfig,
-    Field(discriminator="name"),
+    C.Field(discriminator="name"),
 ]

nshtrainer/lr_scheduler/_base.py

@@ -1,8 +1,9 @@
 import math
 from abc import ABC, abstractmethod
 from collections.abc import Mapping
-from typing import TYPE_CHECKING, ClassVar, Literal, TypeAlias
+from typing import TYPE_CHECKING, Literal
 
+import nshconfig as C
 from lightning.pytorch.utilities.types import (
     LRSchedulerConfigType,
     LRSchedulerTypeUnion,
@@ -10,8 +11,6 @@ from lightning.pytorch.utilities.types import (
 from torch.optim import Optimizer
 from typing_extensions import NotRequired, TypedDict
 
-from ..config import TypedConfig
-
 if TYPE_CHECKING:
     from ..model.base import LightningModuleBase
 
@@ -37,9 +36,7 @@ class LRSchedulerMetadata(TypedDict):
     """Whether to enforce that the monitor exists for reducing the learning rate on plateau. Default is `True`."""
 
 
-class LRSchedulerConfigBase(TypedConfig, ABC):
-    Metadata: ClassVar[TypeAlias] = LRSchedulerMetadata
-
+class LRSchedulerConfigBase(C.Config, ABC):
     @abstractmethod
     def metadata(self) -> LRSchedulerMetadata: ...
 

nshtrainer/lr_scheduler/linear_warmup_cosine.py

@@ -2,12 +2,12 @@ import math
 import warnings
 from typing import Literal
 
+import nshconfig as C
 from torch.optim import Optimizer
 from torch.optim.lr_scheduler import LRScheduler
 from typing_extensions import override
 
-from ..config import Field
-from ._base import LRSchedulerConfigBase
+from ._base import LRSchedulerConfigBase, LRSchedulerMetadata
 
 
 class LinearWarmupCosineAnnealingLR(LRScheduler):
@@ -91,11 +91,11 @@ class LinearWarmupCosineAnnealingLR(LRScheduler):
 class LinearWarmupCosineDecayLRSchedulerConfig(LRSchedulerConfigBase):
     name: Literal["linear_warmup_cosine_decay"] = "linear_warmup_cosine_decay"
 
-    warmup_epochs: int = Field(ge=0)
+    warmup_epochs: int = C.Field(ge=0)
     r"""The number of epochs for the linear warmup phase.
     The learning rate is linearly increased from `warmup_start_lr` to the initial learning rate over this number of epochs."""
 
-    max_epochs: int = Field(gt=0)
+    max_epochs: int = C.Field(gt=0)
     r"""The total number of epochs.
     The learning rate is decayed to `min_lr` over this number of epochs."""
 
@@ -113,7 +113,7 @@ class LinearWarmupCosineDecayLRSchedulerConfig(LRSchedulerConfigBase):
     If `True`, the learning rate will be decayed to `min_lr` over `max_epochs` epochs, and then the learning rate will be increased back to the initial learning rate over `max_epochs` epochs, and so on (this is called a cosine annealing schedule)."""
 
     @override
-    def metadata(self) -> LRSchedulerConfigBase.Metadata:
+    def metadata(self) -> LRSchedulerMetadata:
         return {
             "interval": "step",
         }

nshtrainer/lr_scheduler/reduce_lr_on_plateau.py

@@ -1,12 +1,11 @@
 from typing import TYPE_CHECKING, Literal, cast
 
+from lightning.pytorch.utilities.types import LRSchedulerConfigType
 from torch.optim.lr_scheduler import ReduceLROnPlateau
 from typing_extensions import override
 
-from ll.lr_scheduler._base import LRSchedulerMetadata
-
 from ..model.config import MetricConfig
-from ._base import LRSchedulerConfigBase
+from ._base import LRSchedulerConfigBase, LRSchedulerMetadata
 
 if TYPE_CHECKING:
     from ..model.base import BaseConfig
@@ -43,7 +42,9 @@ class ReduceLROnPlateauConfig(LRSchedulerConfigBase):
     r"""One of `rel`, `abs`. In `rel` mode, dynamic_threshold = best * (1 + threshold) in 'max' mode or best * (1 - threshold) in `min` mode. In `abs` mode, dynamic_threshold = best + threshold in `max` mode or best - threshold in `min` mode. Default: 'rel'."""
 
     @override
-    def create_scheduler_impl(self, optimizer, lightning_module, lr):
+    def create_scheduler_impl(
+        self, optimizer, lightning_module, lr
+    ) -> LRSchedulerConfigType:
         if (metric := self.metric) is None:
             lm_config = cast("BaseConfig", lightning_module.config)
             assert (

nshtrainer/model/__init__.py

@@ -1,7 +1,6 @@
 from typing_extensions import TypeAlias
 
 from .base import Base as Base
-from .base import LightningDataModuleBase as LightningDataModuleBase
 from .base import LightningModuleBase as LightningModuleBase
 from .config import ActSaveConfig as ActSaveConfig
 from .config import BaseConfig as BaseConfig
@@ -33,11 +32,8 @@ from .config import (
 )
 from .config import OptimizationConfig as OptimizationConfig
 from .config import PrimaryMetricConfig as PrimaryMetricConfig
-from .config import PythonLogging as PythonLogging
 from .config import ReproducibilityConfig as ReproducibilityConfig
-from .config import RunnerConfig as RunnerConfig
 from .config import SanityCheckingConfig as SanityCheckingConfig
-from .config import SeedConfig as SeedConfig
 from .config import TrainerConfig as TrainerConfig
 from .config import WandbWatchConfig as WandbWatchConfig
 

nshtrainer/model/base.py

@@ -23,11 +23,12 @@ from .config import (
     EnvironmentLinuxEnvironmentConfig,
     EnvironmentLSFInformationConfig,
     EnvironmentSLURMInformationConfig,
+    EnvironmentSnapshotConfig,
 )
-from .modules.callback import CallbackModuleMixin, CallbackRegistrarModuleMixin
+from .modules.callback import CallbackModuleMixin
 from .modules.debug import DebugModuleMixin
 from .modules.distributed import DistributedMixin
-from .modules.logger import LoggerLightningModuleMixin, LoggerModuleMixin
+from .modules.logger import LoggerLightningModuleMixin
 from .modules.profiler import ProfilerMixin
 from .modules.rlp_sanity_checks import RLPSanityCheckModuleMixin
 from .modules.shared_parameters import SharedParametersModuleMixin
@@ -265,6 +266,9 @@ class LightningModuleBase(  # pyright: ignore[reportIncompatibleMethodOverride]
             boot_time=_try_get(lambda: _psutil().boot_time()),
             load_avg=_try_get(lambda: os.getloadavg()),
         )
+        hparams.environment.snapshot = (
+            EnvironmentSnapshotConfig.from_current_environment()
+        )
 
     def pre_init_update_hparams_dict(self, hparams: MutableMapping[str, Any]):
         """
@@ -309,15 +313,12 @@ class LightningModuleBase(  # pyright: ignore[reportIncompatibleMethodOverride]
     @property
     def datamodule(self):
         datamodule = getattr(self.trainer, "datamodule", None)
-        if datamodule is None:
+        if (datamodule := getattr(self.trainer, "datamodule", None)) is None:
             return None
-
-        if not isinstance(datamodule, LightningDataModuleBase):
+        if not isinstance(datamodule, LightningDataModule):
             raise TypeError(
-                f"datamodule must be a LightningDataModuleBase: {type(datamodule)}"
+                f"datamodule must be a LightningDataModule: {type(datamodule)}"
             )
-
-        datamodule = cast(LightningDataModuleBase[THparams], datamodule)
         return datamodule
 
     if TYPE_CHECKING:
@@ -576,66 +577,3 @@ class LightningModuleBase(  # pyright: ignore[reportIncompatibleMethodOverride]
             "batch": batch,
             "batch_idx": batch_idx,
         }
-
-
-class LightningDataModuleBase(
-    LoggerModuleMixin,
-    CallbackRegistrarModuleMixin,
-    Base[THparams],
-    LightningDataModule,
-    ABC,
-    Generic[THparams],
-):
-    hparams: THparams  # pyright: ignore[reportIncompatibleMethodOverride]
-    hparams_initial: THparams  # pyright: ignore[reportIncompatibleMethodOverride]
-
-    def pre_init_update_hparams_dict(self, hparams: MutableMapping[str, Any]):
-        """
-        Override this method to update the hparams dictionary before it is used to create the hparams object.
-        Mapping-based parameters are passed to the constructor of the hparams object when we're loading the model from a checkpoint.
-        """
-        return hparams
-
-    def pre_init_update_hparams(self, hparams: THparams):
-        """
-        Override this method to update the hparams object before it is used to create the hparams_initial object.
-        """
-        return hparams
-
-    @classmethod
-    def _update_environment(cls, hparams: THparams):
-        hparams.environment.data = _cls_info(cls)
-
-    @override
-    def __init__(self, hparams: THparams):
-        if not isinstance(hparams, BaseConfig):
-            if not isinstance(hparams, MutableMapping):
-                raise TypeError(
-                    f"hparams must be a BaseConfig or a MutableMapping: {type(hparams)}"
-                )
-
-            hparams = self.pre_init_update_hparams_dict(hparams)
-            hparams = self.config_cls().from_dict(hparams)
-        self._update_environment(hparams)
-        hparams = self.pre_init_update_hparams(hparams)
-        super().__init__(hparams)
-
-        self.save_hyperparameters(hparams)
-
-    @property
-    def lightning_module(self):
-        if not self.trainer:
-            raise ValueError("Trainer has not been set.")
-
-        module = self.trainer.lightning_module
-        if not isinstance(module, LightningModuleBase):
-            raise ValueError(
-                f"Trainer's lightning_module is not a LightningModuleBase: {type(module)}"
-            )
-
-        module = cast(LightningModuleBase[THparams], module)
-        return module
-
-    @property
-    def device(self):
-        return self.lightning_module.device

nshtrainer/model/config.py

@@ -1,7 +1,6 @@
 import copy
 import os
 import re
-import signal
 import socket
 import string
 import time
@@ -21,6 +20,7 @@ from typing import (
     runtime_checkable,
 )
 
+import nshconfig as C
 import numpy as np
 import torch
 from lightning.fabric.plugins import CheckpointIO, ClusterEnvironment
@@ -39,7 +39,6 @@ from typing_extensions import Self, TypedDict, TypeVar, override
 from ..callbacks import CallbackConfig
 from ..callbacks.base import CallbackConfigBase
 from ..callbacks.wandb_watch import WandbWatchConfig
-from ..config import Field, TypedConfig
 from ..util.slurm import parse_slurm_node_list
 
 log = getLogger(__name__)
@@ -49,7 +48,7 @@ class IdSeedWarning(Warning):
     pass
 
 
-class BaseProfilerConfig(TypedConfig, ABC):
+class BaseProfilerConfig(C.Config, ABC):
     dirpath: str | Path | None = None
     """
     Directory path for the ``filename``. If ``dirpath`` is ``None`` but ``filename`` is present, the
@@ -200,11 +199,11 @@ class PyTorchProfilerConfig(BaseProfilerConfig):
 
 ProfilerConfig: TypeAlias = Annotated[
     SimpleProfilerConfig | AdvancedProfilerConfig | PyTorchProfilerConfig,
-    Field(discriminator="kind"),
+    C.Field(discriminator="kind"),
 ]
 
 
-class EnvironmentClassInformationConfig(TypedConfig):
+class EnvironmentClassInformationConfig(C.Config):
     name: str
     module: str
     full_name: str
@@ -213,7 +212,7 @@ class EnvironmentClassInformationConfig(TypedConfig):
     source_file_path: Path | None = None
 
 
-class EnvironmentSLURMInformationConfig(TypedConfig):
+class EnvironmentSLURMInformationConfig(C.Config):
     hostname: str
     hostnames: list[str]
     job_id: str
@@ -271,7 +270,7 @@ class EnvironmentSLURMInformationConfig(TypedConfig):
             return None
 
 
-class EnvironmentLSFInformationConfig(TypedConfig):
+class EnvironmentLSFInformationConfig(C.Config):
     hostname: str
     hostnames: list[str]
     job_id: str
@@ -328,7 +327,7 @@ class EnvironmentLSFInformationConfig(TypedConfig):
             return None
 
 
-class EnvironmentLinuxEnvironmentConfig(TypedConfig):
+class EnvironmentLinuxEnvironmentConfig(C.Config):
     """
     Information about the Linux environment (e.g., current user, hostname, etc.)
     """
@@ -347,9 +346,25 @@ class EnvironmentLinuxEnvironmentConfig(TypedConfig):
     load_avg: tuple[float, float, float] | None = None
 
 
-class EnvironmentConfig(TypedConfig):
+class EnvironmentSnapshotConfig(C.Config):
+    snapshot_dir: Path | None = None
+    modules: list[str] | None = None
+
+    @classmethod
+    def from_current_environment(cls):
+        draft = cls.draft()
+        if snapshot_dir := os.environ.get("NSHRUNNER_SNAPSHOT_DIR"):
+            draft.snapshot_dir = Path(snapshot_dir)
+        if modules := os.environ.get("NSHRUNNER_SNAPSHOT_MODULES"):
+            draft.modules = modules.split(",")
+        return draft.finalize()
+
+
+class EnvironmentConfig(C.Config):
     cwd: Path | None = None
 
+    snapshot: EnvironmentSnapshotConfig | None = None
+
     python_executable: Path | None = None
     python_path: list[Path] | None = None
     python_version: str | None = None
@@ -372,7 +387,7 @@ class EnvironmentConfig(TypedConfig):
     seed_workers: bool | None = None
 
 
-class BaseLoggerConfig(TypedConfig, ABC):
+class BaseLoggerConfig(C.Config, ABC):
     enabled: bool = True
     """Enable this logger."""
 
@@ -426,7 +441,7 @@ def _wandb_available():
 class WandbLoggerConfig(CallbackConfigBase, BaseLoggerConfig):
     kind: Literal["wandb"] = "wandb"
 
-    enabled: bool = Field(default_factory=lambda: _wandb_available())
+    enabled: bool = C.Field(default_factory=lambda: _wandb_available())
     """Enable WandB logging."""
 
     priority: int = 2
@@ -543,7 +558,7 @@ def _tensorboard_available():
 class TensorboardLoggerConfig(BaseLoggerConfig):
     kind: Literal["tensorboard"] = "tensorboard"
 
-    enabled: bool = Field(default_factory=lambda: _tensorboard_available())
+    enabled: bool = C.Field(default_factory=lambda: _tensorboard_available())
     """Enable TensorBoard logging."""
 
     priority: int = 2
@@ -589,7 +604,7 @@ class TensorboardLoggerConfig(BaseLoggerConfig):
 
 LoggerConfig: TypeAlias = Annotated[
     WandbLoggerConfig | CSVLoggerConfig | TensorboardLoggerConfig,
-    Field(discriminator="kind"),
+    C.Field(discriminator="kind"),
 ]
 
 
@@ -684,7 +699,7 @@ class LoggingConfig(CallbackConfigBase):
             yield from logger.construct_callbacks(root_config)
 
 
-class GradientClippingConfig(TypedConfig):
+class GradientClippingConfig(C.Config):
     enabled: bool = True
     """Enable gradient clipping."""
     value: int | float
@@ -719,41 +734,6 @@ class OptimizationConfig(CallbackConfigBase):
         ).construct_callbacks(root_config)
 
 
-LogLevel: TypeAlias = Literal[
-    "CRITICAL", "FATAL", "ERROR", "WARN", "WARNING", "INFO", "DEBUG"
-]
-
-
-class PythonLogging(TypedConfig):
-    log_level: LogLevel | None = None
-    """Log level to use for the Python logger (or None to use the default)."""
-
-    rich: bool = False
-    """If enabled, will use the rich library to format the Python logger output."""
-    rich_tracebacks: bool = True
-    """If enabled, will use the rich library to format the Python logger tracebacks."""
-
-    lovely_tensors: bool = False
-    """If enabled, will use the lovely-tensors library to format PyTorch tensors. False by default as it causes issues when used with `torch.vmap`."""
-    lovely_numpy: bool = False
-    """If enabled, will use the lovely-numpy library to format numpy arrays. False by default as it causes some issues with other libaries."""
-
-    def pretty_(
-        self,
-        *,
-        log_level: LogLevel | None = "INFO",
-        torch: bool = True,
-        numpy: bool = True,
-        rich: bool = True,
-        rich_tracebacks: bool = True,
-    ):
-        self.log_level = log_level
-        self.lovely_tensors = torch
-        self.lovely_numpy = numpy
-        self.rich = rich
-        self.rich_tracebacks = rich_tracebacks
-
-
 TPlugin = TypeVar(
     "TPlugin",
     Precision,
@@ -813,7 +793,7 @@ StrategyLiteral: TypeAlias = Literal[
 ]
 
 
-class CheckpointLoadingConfig(TypedConfig):
+class CheckpointLoadingConfig(C.Config):
     path: Literal["best", "last", "hpc"] | str | Path | None = None
     """
     Checkpoint path to use when loading a checkpoint.
@@ -825,7 +805,7 @@ class CheckpointLoadingConfig(TypedConfig):
     """
 
 
-class DirectoryConfig(TypedConfig):
+class DirectoryConfig(C.Config):
     project_root: Path | None = None
     """
     Root directory for this project.
@@ -901,7 +881,7 @@ class DirectoryConfig(TypedConfig):
         return log_dir
 
 
-class ReproducibilityConfig(TypedConfig):
+class ReproducibilityConfig(C.Config):
     deterministic: bool | Literal["warn"] | None = None
     """
     If ``True``, sets whether PyTorch operations must use deterministic algorithms.
@@ -1116,7 +1096,7 @@ CheckpointCallbackConfig: TypeAlias = Annotated[
     ModelCheckpointCallbackConfig
     | LatestEpochCheckpointCallbackConfig
     | OnExceptionCheckpointCallbackConfig,
-    Field(discriminator="kind"),
+    C.Field(discriminator="kind"),
 ]
 
 
@@ -1514,7 +1494,7 @@ class ActSaveConfig(CallbackConfigBase):
         return [ActSaveCallback()]
 
 
-class SanityCheckingConfig(TypedConfig):
+class SanityCheckingConfig(C.Config):
     reduce_lr_on_plateau: Literal["disable", "error", "warn"] = "error"
     """
     If enabled, will do some sanity checks if the `ReduceLROnPlateau` scheduler is used:
@@ -1524,7 +1504,7 @@ class SanityCheckingConfig(TypedConfig):
     """
 
 
-class TrainerConfig(TypedConfig):
+class TrainerConfig(C.Config):
     checkpoint_loading: CheckpointLoadingConfig = CheckpointLoadingConfig()
     """Checkpoint loading configuration options."""
 
@@ -1739,87 +1719,7 @@ class TrainerConfig(TypedConfig):
     """If enabled, will set the torch float32 matmul precision to the specified value. Useful for faster training on Ampere+ GPUs."""
 
 
-class SeedConfig(TypedConfig):
-    seed: int
-    """Seed for the random number generator."""
-
-    seed_workers: bool = False
-    """Whether to seed the workers of the dataloader."""
-
-
-Signal: TypeAlias = Literal[
-    "SIGHUP",
-    "SIGINT",
-    "SIGQUIT",
-    "SIGILL",
-    "SIGTRAP",
-    "SIGABRT",
-    "SIGBUS",
-    "SIGFPE",
-    "SIGKILL",
-    "SIGUSR1",
-    "SIGSEGV",
-    "SIGUSR2",
-    "SIGPIPE",
-    "SIGALRM",
-    "SIGTERM",
-    "SIGCHLD",
-    "SIGCONT",
-    "SIGSTOP",
-    "SIGTSTP",
-    "SIGTTIN",
-    "SIGTTOU",
-    "SIGURG",
-    "SIGXCPU",
-    "SIGXFSZ",
-    "SIGVTALRM",
-    "SIGPROF",
-    "SIGWINCH",
-    "SIGIO",
-    "SIGPWR",
-    "SIGSYS",
-    "SIGRTMIN",
-    "SIGRTMAX",
-]
-
-
-class SubmitConfig(TypedConfig):
-    auto_requeue_signals: list[Signal] = [
-        # "SIGUSR1",
-        # On SIGURG:
-        # Important note from https://amrex-astro.github.io/workflow/olcf-workflow.html:
-        # We can also ask the job manager to send a warning signal some amount of time before the allocation expires by passing -wa 'signal' and -wt '[hour:]minute' to bsub. We can then have bash create a dump_and_stop file when it receives the signal, which will tell Castro to output a checkpoint file and exit cleanly after it finishes the current timestep. An important detail that I couldn't find documented anywhere is that the job manager sends the signal to all the processes in the job, not just the submission script, and we have to use a signal that is ignored by default so Castro doesn't immediately crash upon receiving it. SIGCHLD, SIGURG, and SIGWINCH are the only signals that fit this requirement and of these, SIGURG is the least likely to be triggered by other events.
-        "SIGURG"
-    ]
-    """Signals that will trigger an automatic requeue of the job."""
-
-    def _resolved_auto_requeue_signals(self) -> list[signal.Signals]:
-        return [getattr(signal.Signals, sig) for sig in self.auto_requeue_signals]
-
-
-class RunnerConfig(TypedConfig):
-    python_logging: PythonLogging = PythonLogging()
-    """Python logging configuration options."""
-
-    seed: SeedConfig = SeedConfig(seed=0)
-    """Seed everything configuration options."""
-
-    submit: SubmitConfig = SubmitConfig()
-    """Submit (e.g., SLURM or LSF) configuration options."""
-
-    dump_run_information: bool = True
-    """
-    If enabled, will dump different bits of run information to the output directory before starting the run.
-    This includes:
-        - Run config
-        - Full set of environment variables
-    """
-
-    additional_env_vars: dict[str, str] = {}
-    """Additional environment variables to set when running the script."""
-
-
-class MetricConfig(TypedConfig):
+class MetricConfig(C.Config):
     name: str
     """The name of the primary metric."""
 
@@ -1851,8 +1751,8 @@ class MetricConfig(TypedConfig):
 PrimaryMetricConfig: TypeAlias = MetricConfig
 
 
-class BaseConfig(TypedConfig):
-    id: str = Field(default_factory=lambda: BaseConfig.generate_id())
+class BaseConfig(C.Config):
+    id: str = C.Field(default_factory=lambda: BaseConfig.generate_id())
     """ID of the run."""
     name: str | None = None
     """Run name."""
@@ -1867,15 +1767,13 @@ class BaseConfig(TypedConfig):
 
     debug: bool = False
     """Whether to run in debug mode. This will enable debug logging and enable debug code paths."""
-    environment: Annotated[EnvironmentConfig, Field(repr=False)] = EnvironmentConfig()
+    environment: Annotated[EnvironmentConfig, C.Field(repr=False)] = EnvironmentConfig()
     """A snapshot of the current environment information (e.g. python version, slurm info, etc.). This is automatically populated by the run script."""
 
     directory: DirectoryConfig = DirectoryConfig()
     """Directory configuration options."""
     trainer: TrainerConfig = TrainerConfig()
     """PyTorch Lightning trainer configuration options. Check Lightning's `Trainer` documentation for more information."""
-    runner: RunnerConfig = RunnerConfig()
-    """`ll.Runner` configuration options."""
 
     primary_metric: PrimaryMetricConfig | None = None
     """Primary metric configuration options. This is used in the following ways: