PyTorch Geometric Adapter

PyG conversion helpers from annnet.adapters.pyg_adapter.

annnet.adapters.pyg_adapter

Classes

Functions

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 (polars/pandas via _df_to_rows)
• Heterogeneous (vertex kinds)
• Hypergraph-safe (reification)
• Slice-aware (boolean masks)