Skip to content

PyTorch Geometric Adapter

PyG conversion helpers from annnet.adapters.pyg_adapter.

annnet.adapters.pyg_adapter

AnnNet-PyTorch Geometric adapter for AnnNet.

Provides: to_pyg(G) -> torch_geometric.data.HeteroData

PyTorch Geometric represents graph data as tensors. This adapter exports AnnNet vertices and edges into a heterogeneous graph structure suitable for downstream GNN workflows.

AnnNet-specific structures such as slices, multilayer metadata, hyperedge semantics, and rich attribute tables are only exported where they can be mapped to tensor-compatible node, edge, or graph-level fields.

Classes

Functions

to_pyg 
to_pyg(
    graph,
    node_features=None,
    edge_features=None,
    slice_id=None,
    hyperedge_mode="reify",
    device="cpu",
)

Direct AnnNet -> PyTorch Geometric adapter.

  • Respects AnnNet architecture (uses entity_types, edge_definitions)
  • Narwhals-compatible dataframe input via shared row conversion
  • Heterogeneous (vertex kinds)
  • Hypergraph-safe (reification)
  • Slice-aware (boolean masks)