From 17864d33f99cadbf288ec3bc6d89fb2339c50505 Mon Sep 17 00:00:00 2001 From: Uri Granta Date: Thu, 25 Jan 2024 11:51:34 +0000 Subject: [PATCH] Fix types_old --- trieste/ask_tell_optimization.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/trieste/ask_tell_optimization.py b/trieste/ask_tell_optimization.py index 367dad09ad..46053f66b0 100644 --- a/trieste/ask_tell_optimization.py +++ b/trieste/ask_tell_optimization.py @@ -363,13 +363,13 @@ def from_record( # so the model was already trained # thus there is no need to train it again - # type ignore below is due to the fact that overloads don't allow - # optional acquisition_rule along with acquisition_state - return cls( + # type ignore below is because this relies on subclasses not overriding __init__ + # ones that do may also need to override this to get it to work + return cls( # type: ignore search_space, record.datasets, record.models, - acquisition_rule=acquisition_rule, # type: ignore + acquisition_rule=acquisition_rule, acquisition_state=record.acquisition_state, fit_model=False, )