metapruning/__init__.py

"""
MetaPruning: Meta Pruning via Graph Metanetworks
Paper: "Meta Pruning via Graph Metanetworks: A Universal Meta-Learning Framework for Network Pruning"
arXiv:2506.12041

Modules:
    graph:     Network <-> Graph bijection
    gnn:       PNAConv-based metanetwork
    train_metanetwork: Meta-training loop
    inference: Pruning inference pipeline
"""

from .graph import resnet_to_graph, graph_to_resnet, create_transformed_model, Graph
from .gnn import Metanetwork, PNAConvLayer, EdgeInvertor

__all__ = [
    "resnet_to_graph",
    "graph_to_resnet", 
    "create_transformed_model",
    "Graph",
    "Metanetwork",
    "PNAConvLayer",
    "EdgeInvertor",
]