dmlc--dgl
067cd74420
* first commit * update * huh * fix * update * revert core * fix * update * rewrite * oops * address comments * add graph name * address comments * remove sample metadata file * address comments * fix * remove * add docs
12 行
236 B
Python
12 行
236 B
Python
REGISTRY = {}
|
|
|
|
def register_array_parser(name):
|
|
def _deco(cls):
|
|
REGISTRY[name] = cls
|
|
return cls
|
|
return _deco
|
|
|
|
def get_array_parser(**fmt_meta):
|
|
cls = REGISTRY[fmt_meta.pop('name')]
|
|
return cls(**fmt_meta)
|