PyTorch Geometric Adapter

PyG conversion helpers from annnet.adapters.pyg_adapter.

annnet.adapters.pyg_adapter

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)