2
0
mirror of https://github.com/m-labs/artiq.git synced 2024-12-25 11:18:27 +08:00

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

This commit is contained in:
Sebastien Bourdeauducq 2018-05-02 12:32:35 +08:00
parent 8d66d53e06
commit 8c69d939fb

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