From 1b8b649878705d76cc23166cddf3009869a23ef7 Mon Sep 17 00:00:00 2001
From: gehbiszumeis <16896724+gehbiszumeis@users.noreply.github.com>
Date: Fri, 29 Oct 2021 14:04:20 +0200
Subject: [PATCH] fix: rework Y_metadata part for compatibility with unittests
 !minor

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

diff --git a/GPy/core/gp.py b/GPy/core/gp.py
index 62fd43b7..3cfa38ec 100644
--- a/GPy/core/gp.py
+++ b/GPy/core/gp.py
@@ -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: