dmlc--dgl
Quan (Andy) Gan
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
18 行
444 B
Python
18 行
444 B
Python
import os
|
|
from contextlib import contextmanager
|
|
import logging
|
|
from numpy.lib.format import open_memmap
|
|
|
|
@contextmanager
|
|
def setdir(path):
|
|
try:
|
|
os.makedirs(path, exist_ok=True)
|
|
cwd = os.getcwd()
|
|
logging.info('Changing directory to %s' % path)
|
|
logging.info('Previously: %s' % cwd)
|
|
os.chdir(path)
|
|
yield
|
|
finally:
|
|
logging.info('Restoring directory to %s' % cwd)
|
|
os.chdir(cwd)
|