src.neurodags.nodes¶
Submodules¶
Functions¶
|
Register a callable node factory. |
|
Return a registered node by name. |
|
Yield registered nodes as |
|
Return the registered node names sorted alphabetically. |
|
Remove all registered nodes (primarily useful for tests). |
|
Remove a node from the registry if present. |
|
Import all modules in the package to trigger registrations. |
|
Package Contents¶
- src.neurodags.nodes.register_node(func: NodeCallable | None = None, *, name: str | None = None, override: bool = False) NodeCallable¶
Register a callable node factory.
Can be used as
@register_nodeor@register_node(name="alias").
- src.neurodags.nodes.get_node(name: str) NodeCallable¶
Return a registered node by name.
- src.neurodags.nodes.iter_nodes() collections.abc.Iterable[tuple[str, NodeCallable]]¶
Yield registered nodes as
(name, callable)pairs.
- src.neurodags.nodes.list_nodes() tuple[str, Ellipsis]¶
Return the registered node names sorted alphabetically.
- src.neurodags.nodes.clear_node_registry() None¶
Remove all registered nodes (primarily useful for tests).
- src.neurodags.nodes.unregister_node(name: str) None¶
Remove a node from the registry if present.
- src.neurodags.nodes.discover(package: str | None = None) None¶
Import all modules in the package to trigger registrations.
- src.neurodags.nodes.register_node_with_name(name: str, func: collections.abc.Callable) None¶