12 lines
368 B
Python
12 lines
368 B
Python
import importlib
|
|
import pkgutil
|
|
import os
|
|
|
|
from core.topo import Topo, TopoConfig
|
|
|
|
pkg_dir = os.path.dirname(__file__)
|
|
for (module_loader, name, ispkg) in pkgutil.iter_modules([pkg_dir]):
|
|
importlib.import_module('.' + name, __package__)
|
|
|
|
TOPO_CONFIGS = {cls.NAME: cls for cls in TopoConfig.__subclasses__()}
|
|
TOPOS = {cls.NAME: cls for cls in Topo.__subclasses__()} |