baybe.surrogates.utils.scale_model¶

baybe.surrogates.utils.scale_model(model_cls: type[Surrogate])[source]¶

Wrap a Surrogate class such that it operates with scaled representations.

Parameters:: model_cls (type[Surrogate]) – A Surrogate model class that should be wrapped.
Returns:: A wrapped version of the class.