liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 00:06:58 +08:00
Code Activity

fix picker error on importlib loaded module

Browse Source
This commit is contained in:
Dong Zhou
2021-05-18 22:12:41 +08:00
parent 19eda8f4f0
commit c72ee9091e
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
qlib/utils/__init__.py
View File

@@ -8,6 +8,7 @@ from __future__ import print_function
import os import os
import pickle import pickle
import re import re
import sys
import copy import copy
import json import json
import yaml import yaml
@@ -178,8 +179,10 @@ def get_module_by_module_path(module_path: Union[str, ModuleType]):
module = module_path module = module_path
else: else:
if module_path.endswith(".py"): if module_path.endswith(".py"):
module_spec = importlib.util.spec_from_file_location("", module_path) module_name = "dummy"
module_spec = importlib.util.spec_from_file_location(module_name, module_path)
module = importlib.util.module_from_spec(module_spec) module = importlib.util.module_from_spec(module_spec)
sys.modules[module_name] = module
module_spec.loader.exec_module(module) module_spec.loader.exec_module(module)
else: else:
module = importlib.import_module(module_path) module = importlib.import_module(module_path)