liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-04 11:30:57 +08:00
Code Activity

Merge pull request #431 from evanzd/fix_get_module

Browse Source 
Fix `get_module_by_module_path` to support pickle module loaded from arbitrage source file
This commit is contained in:
you-n-g
2021-05-21 12:53:14 +08:00
committed by GitHub
parent 19eda8f4f0 2fa7ef32fb
commit 3d71fd1966
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 pickle
import re
import sys
import copy
import json
import yaml
@@ -178,8 +179,10 @@ def get_module_by_module_path(module_path: Union[str, ModuleType]):
module = module_path
else:
if module_path.endswith(".py"):
module_spec = importlib.util.spec_from_file_location("", module_path)
module_name = re.sub("^[^a-zA-Z_]+", "", re.sub("[^0-9a-zA-Z_]", "", module_path[:-3].replace("/", "_")))
module_spec = importlib.util.spec_from_file_location(module_name, module_path)
module = importlib.util.module_from_spec(module_spec)
sys.modules[module_name] = module
module_spec.loader.exec_module(module)
else:
module = importlib.import_module(module_path)