add: built_from_dict method for White Kernel

SheffieldML · Dec 1, 2021 · 23aae96 · 23aae96
1 parent c84d968
commit 23aae96
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/GPy/kern/src/static.py b/GPy/kern/src/static.py
@@ -68,6 +68,11 @@ def to_dict(self):
         input_dict = super(White, self)._save_to_input_dict()
         input_dict["class"] = "GPy.kern.White"
         return input_dict
+
+    @staticmethod
+    def _build_from_input_dict(kernel_class, input_dict):
+        useGPU = input_dict.pop('useGPU', None)
+        return White(**input_dict)
 
     def K(self, X, X2=None):
         if X2 is None: