Skip to content

Build

build

Functions to build a dependency graph.

Functions

build_dependency_graph

build_dependency_graph(parameters_config: dict[str, Any]) -> DiGraph

Build a dependency graph.

Build a directed graph here:

  • nodes = parameter names
  • edge A → B means: "to compute/sample B, you need A first"

Parameters:

Name Type Description Default
parameters_config dict[str, Any]

A dictionary of parameters.

required

Returns:

Type Description
DiGraph

A DiGraph for topological sort.

add_dependencies_to_graph

add_dependencies_to_graph(graph: DiGraph, dependencies: set[str], parameter_name: str) -> None

Add dependencies of a parameter to the graph.

Parameters:

Name Type Description Default
graph DiGraph

Dependency graph.

required
dependencies set[str]

A set of dependencies.

required
parameter_name str

Parameter name.

required