secondmind-labs / trieste

A Bayesian optimization toolbox built on TensorFlow
Apache License 2.0
212 stars 41 forks source link

Changing KERNEL_LENGTHSCALE to tf.constant results in serialisation failure in expected_improvement.pct.py #833

Open uri-granta opened 3 months ago

uri-granta commented 3 months ago

Changing KERNEL_LENGTHSCALE from tf.cast to tf.constant results in the documentation build failing during serialisation (hitting an assert id(obj) not in self.memo in pickle.py while using dill). This does not happen in the quickdocs run and was not easy to reproduce in the integration tests.

Low priority but would be good to get to the bottom of this.