CodeEntropy.levels.nodes.detect_levels module

Detect hierarchy levels present for each molecule in the reduced universe.

This module defines a static DAG node responsible for determining which hierarchical levels (united_atom, residue, polymer) apply to each molecule.

class CodeEntropy.levels.nodes.detect_levels.DetectLevelsNode[source]

Bases: object

Static node that determines hierarchy levels per molecule.

Produces:

shared_data[“levels”] shared_data[“number_molecules”]

run(shared_data: dict[str, Any]) dict[str, Any][source]

Detect levels and store results in shared_data.

Parameters:

shared_data – Shared data dictionary. Requires: - “reduced_universe”

Returns:

  • “levels”: List of levels per molecule.

  • ”number_molecules”: Total molecule count.

Return type:

Dict containing

Raises:

KeyError – If required keys are missing.