fix: rework Y_metadata part for compatibility with unittests !minor

This commit is contained in:
gehbiszumeis 2021-10-29 14:04:20 +02:00
parent a133947b42
commit 1b8b649878

View file

@ -164,9 +164,11 @@ class GP(Model):
input_dict["inference_method"] = GPy.inference.latent_function_inference.LatentFunctionInference.from_dict(input_dict["inference_method"])
# converts Y_metadata from serializable to array. We should create a Metadata class
# Y_metadata = input_dict.get("Y_metadata")
# input_dict["Y_metadata"] = Y_metadata
input_dict["Y_metadata"] = {k: np.array(input_dict['Y_metadata'][k]) for k in input_dict['Y_metadata'].keys()}
Y_metadata = input_dict.get("Y_metadata")
if isinstance(Y_metadata, dict):
input_dict["Y_metadata"] = {k: np.array(Y_metadata[k]) for k in Y_metadata.keys()}
else:
input_dict["Y_metadata"] = Y_metadata
normalizer = input_dict.get("normalizer")
if normalizer is not None: