From 4de3fdda20abcea2bba678c25eb802686ab47b70 Mon Sep 17 00:00:00 2001
From: gehbiszumeis <16896724+gehbiszumeis@users.noreply.github.com>
Date: Wed, 27 Oct 2021 16:05:24 +0200
Subject: [PATCH] fix: made Y_metadata dict content serializable

---
 GPy/core/gp.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/GPy/core/gp.py b/GPy/core/gp.py
index 0704c6a6..5adc8ce4 100644
--- a/GPy/core/gp.py
+++ b/GPy/core/gp.py
@@ -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