apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-06-08 15:05:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: made Y_metadata dict content serializable

Browse source
This commit is contained in:
gehbiszumeis 2021-10-27 16:05:24 +02:00
parent fcb43ce5dd
commit 4de3fdda20
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
GPy/core/gp.py
View file

@ -134,9 +134,10 @@ class GP(Model):
if self.mean_function is not None:
input_dict["mean_function"] = self.mean_function.to_dict()
input_dict["inference_method"] = self.inference_method.to_dict()
#FIXME: Assumes the Y_metadata is serializable. We should create a Metadata class
# TODO: We should create a Metadata class
if self.Y_metadata is not None:
input_dict["Y_metadata"] = self.Y_metadata
# make Y_metadata serializable
input_dict["Y_metadata"] = {k: self.Y_metadata[k].tolist() for k in self.Y_metadata.keys()}
if self.normalizer is not None:
input_dict["normalizer"] = self.normalizer.to_dict()
return input_dict