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

pull/1017/head
Sebastien Bourdeauducq 2018-05-02 12:32:35 +08:00
parent 8d66d53e06
commit 8c69d939fb
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