secondmind-labs / trieste

A Bayesian optimization toolbox built on TensorFlow
Apache License 2.0
219 stars 42 forks source link

Fix slow tests #831

Closed uri-granta closed 6 months ago

uri-granta commented 6 months ago

Related issue(s)/PRs:

Summary

Remove slow deep ensemble test with float32 as randomness has this currently failing (plus it's fairly slow). We still have the other faster tests ensuring that float32 isn't broken.

Perplexedly replace one tf.constant with a tf.cast as the latter was resulting in a result.save error in the expected improvements notebook on line 298 (but not when run in quick mode)!

Fully backwards compatible: yes

PR checklist