diff --git a/baybe/surrogates/gaussian_process/presets/__init__.py b/baybe/surrogates/gaussian_process/presets/__init__.py index c4d2f79b6c..76d79c3af3 100644 --- a/baybe/surrogates/gaussian_process/presets/__init__.py +++ b/baybe/surrogates/gaussian_process/presets/__init__.py @@ -6,10 +6,12 @@ ) from baybe.surrogates.gaussian_process.presets.default import DefaultKernelFactory from baybe.surrogates.gaussian_process.presets.edbo import EDBOKernelFactory +from baybe.surrogates.gaussian_process.presets.chen import CHENKernelFactory __all__ = [ "DefaultKernelFactory", "EDBOKernelFactory", "make_gp_from_preset", "GaussianProcessPreset", + "CHENKernelFactory", ]