secondmind-labs / trieste

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

Support initialising AskTellOptimizer with extended datasets #849

Closed uri-granta closed 1 month ago

uri-granta commented 1 month ago

Related issue(s)/PRs:

Summary

Another use case for track_data=False: being able to initialise a new AskTellOptimizer using the state saved after ask was called but before the datasets were updated. The acquisition state is already updated correctly due to filter_dataset. This fix adds the dataset length to the AskTellOptimiserState so that the local_data_ixs can also be updated automatically.

Fully backwards compatible: yes

PR checklist