from .wangnet import WangNet
from .graphgen import GraphGen

MODEL_DICT = dict(
    w=WangNet,
    g=GraphGen,
)

from .wangnet import WangNet
from .graphgen import GraphGen

MODEL_DICT = dict(
    w=WangNet,
    g=GraphGen,
)
