Match base class rename

secondmind-labs · Aug 25, 2023 · 9ee2d21 · 9ee2d21
1 parent 8a2cba9
commit 9ee2d21
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/trieste/acquisition/rule.py b/trieste/acquisition/rule.py
@@ -1411,7 +1411,7 @@ def get_initialize_subspaces_mask(
         return tf.logical_not(get_unique_points_mask(centres, tolerance=1e-6))
 
 
-class TREGOBox(SingleObjectiveTRBox):
+class TREGOBox(SingleObjectiveTrustRegionBox):
     """
     A trust region box algorithm that alternates between regular EGO steps and local steps within a
     trust region.