1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-14 08:16:54 +08:00

Fix the multi-processing bug

This commit is contained in:
Young
2021-04-12 06:33:31 +00:00
parent cca43cf102
commit b15e5e33fd
3 changed files with 8 additions and 8 deletions

View File

@@ -1,10 +1,9 @@
from qlib.model.ens.ensemble import Ensemble, RollingEnsemble
from typing import Callable, Union
from qlib.utils.serial import Serializable
from joblib import Parallel, delayed
class Group(Serializable):
class Group:
"""Group the objects based on dict"""
def __init__(self, group_func=None, ens: Ensemble = None):
@@ -45,9 +44,8 @@ class Group(Serializable):
dict: grouped_dict like {G1: object, G2: object}
"""
# FIXME: The multiprocessing will raise the following error
# NotImplementedError: Please specify valid `_ens_func`.
# The problem maybe the state of the function is lost
# NOTE: The multiprocessing will raise error if you use `Serializable`
# Because the `Serializable` will affect the behaviours of pickle
grouped_dict = self.group(ungrouped_dict, *args, **kwargs)
key_l = []