1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-02 10:31:00 +08:00

fix picker error on importlib loaded module

This commit is contained in:
Dong Zhou
2021-05-18 22:12:41 +08:00
parent 19eda8f4f0
commit c72ee9091e

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 = "dummy"
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)