This release is a pre-release and may not be stable for production use.
nshconfig
Typed, provenance-aware configuration for ML runs, powered by Pydantic.
One verb family and one value:
Cls.config_draft()gives a mutable draft: plain Python assignment, nested configs auto-create, validation deferred.C.interp(lambda c: ...)is a value that resolves against the config tree at validation. It is legal anywhere a value sits: assigned on a draft, inside amodel_validatedict, or as a class default. This is Hydra-style interpolation, in Python, mostly type-checked.draft.config_finalize()resolves interpolation, validates once, and returns a frozen, hashable, fully-concrete config.
Plus provenance: final.config_explain("optim.lr") answers "why did this run use that value?" down
to file, line, function, source text, and the interpolation's "because" chain.
Install
pip install nshconfig # pydantic>=2.13, Python>=3.10
pip install nshconfig[treescope] # optional rich notebook rendering
Quickstart
import nshconfig as C
class LNConfig(C.Config):
dim: int = 32 # plain default; leaf classes need no interpolation
eps: float = 1e-5
class EncoderConfig(C.Config):
ln: LNConfig
class HeadConfig(C.Config):
# class-level interpolation: a value sitting in the default slot
dim: int = C.interp(lambda c: c.nearest(ModelConfig).dim)
class ModelConfig(C.Config):
dim: int = 768
encoder: EncoderConfig
head: HeadConfig
class TrainConfig(C.Config):
batch: int = 8
model: ModelConfig
# compose in a notebook; helpers are plain functions (your "config groups")
def large(cfg: TrainConfig) -> None:
cfg.model.dim = 1024
cfg = TrainConfig.config_draft()
large(cfg)
# instance-level interpolation: wire THIS tree only, at composition time
cfg.model.encoder.ln.dim = C.interp(lambda c: c.nearest(ModelConfig).dim)
final = cfg.config_finalize() # the one validation boundary
assert final.model.encoder.ln.dim == 1024 # followed the knob
assert final.model.head.dim == 1024 # class-default rule, same machinery
final.model_dump_json() # concrete values only: the run record
# why did this run use that value?
print(final.config_explain("model.head.dim"))
# model.head.dim = 1024
# interpolated to 1024 by <lambda> @ configs.py:11 (class default)
# because model.dim = 1024
# class-default rule: interp(<<lambda> @ configs.py:11>) (active)
Explicit always beats interpolation (presence in the input slot beats the default slot; last
write wins; del re-arms). Nothing pending can reach a final, a dump, an f-string, or an if
without a loud error naming the dotted path and the source line. Drafts cloudpickle to clusters
mid-composition and finalize on the far side, provenance included.
The Ctx API (what the lambda sees)
| Accessor | Hydra equivalent | Sees |
|---|---|---|
c.self() / c.self(Cls) |
same level | own fields, earlier markers already resolved |
c.parent() / c.parent(Cls) |
${..x} |
one level up, resolved |
c.parent(n) / c.parent(n, Cls) |
${...x} |
exactly n ancestor hops up |
c.root() / c.root(Cls) |
${a.b} |
the validation root, incl. sibling subtrees |
c.nearest(Cls) |
(none: better) | nearest enclosing Cls; survives restructuring |
Passing a class gives typed field access and a runtime assertion. Omitting the class keeps the selector dynamic and performs no type assertion.
License
MIT
1 maintainer
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file nshconfig-2.0.0a2.tar.gz.
File metadata
- Download URL: nshconfig-2.0.0a2.tar.gz
- Upload date:
- Size: 16.8 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.9.18 {"installer":{"name":"uv","version":"0.9.18","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
a2303601d682d2315dbd581b580584d8de62a1513345a7dd4e79393a934bd723
|
|
| MD5 |
82ebc1c09b774b043a1d6173937c1a10
|
|
| BLAKE2b-256 |
ef42a407a3c1fe81781dbe5daec7155903fc75df783ea7f294d52a294b69da44
|
File details
Details for the file nshconfig-2.0.0a2-py3-none-any.whl.
File metadata
- Download URL: nshconfig-2.0.0a2-py3-none-any.whl
- Upload date:
- Size: 21.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: uv/0.9.18 {"installer":{"name":"uv","version":"0.9.18","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"Ubuntu","version":"24.04","id":"noble","libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
09312c90c6c3cceabc4ef3962b3a390a99df1d23c1d74448f45ce4b0214f744f
|
|
| MD5 |
7df731b2ebf575e1810e3a1540261900
|
|
| BLAKE2b-256 |
bcaae31f6b830ee8a7d32abd147ad7b68cbc4edf11135225aa667517967b816d
|
