worker: restore sys.modules in examine() (#976)

This commit is contained in:
Sebastien Bourdeauducq 2018-05-02 12:32:35 +08:00
parent b84a2a1eeb
commit 6a0bba1d89
1 changed files with 6 additions and 1 deletions

View File

@ -4,6 +4,7 @@ import os
import logging
import traceback
from collections import OrderedDict
from copy import copy
import h5py
@ -152,7 +153,11 @@ class ExamineDatasetMgr:
def examine(device_mgr, dataset_mgr, file):
module = file_import(file)
previous_modules = copy(sys.modules)
try:
module = file_import(file)
finally:
sys.modules = previous_modules
for class_name, exp_class in module.__dict__.items():
if class_name[0] == "_":
continue