diff --git a/bofire/utils/torch_tools.py b/bofire/utils/torch_tools.py index 8932334a..e43cf6fe 100644 --- a/bofire/utils/torch_tools.py +++ b/bofire/utils/torch_tools.py @@ -534,7 +534,12 @@ def _x_adapt(feat): continue if allowed_objectives is not None: - if not any([isinstance(feat.objective, allowed_objective_) for allowed_objective_ in allowed_objectives]): + if not any( + [ + isinstance(feat.objective, allowed_objective_) + for allowed_objective_ in allowed_objectives + ] + ): continue callables.append(