Hongzhi (Steve), Chen
a208e8868b
* [Misc] Black auto fix. * fix pylint disable Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
18 lines
377 B
Python
18 lines
377 B
Python
"""dgl distributed.optims."""
|
|
import importlib
|
|
import os
|
|
import sys
|
|
|
|
from ...backend import backend_name
|
|
from ...utils import expand_as_pair
|
|
|
|
|
|
def _load_backend(mod_name):
|
|
mod = importlib.import_module(".%s" % mod_name, __name__)
|
|
thismod = sys.modules[__name__]
|
|
for api, obj in mod.__dict__.items():
|
|
setattr(thismod, api, obj)
|
|
|
|
|
|
_load_backend(backend_name)
|