get_clusters

samplomatic.graph_utils.get_clusters(graph: PyDiGraph[T, S], filter_fn: Callable[[T], bool]) → Iterator[list[int]][source]

Yield lists of nodes that all pass the filter and are connected in the graph.

Parameters:

  • graph – The graph to search.

  • filter_fn – A filter function acting on node data that returns booleans.

Yields:

Lists of nodes, where each list is a maximal weakly connected set of nodes that all pass the filter.