From 6070764ba93515e770b5e298fb3f38827ba23120 Mon Sep 17 00:00:00 2001 From: Elizabeth Santorella Date: Sat, 24 Aug 2024 05:36:06 -0700 Subject: [PATCH] Take out error about HSS not being supported with batch trials Summary: Context: This error seems to be wrong. This seems to be supported, and we can't think why it wouldn't be. Reviewed By: saitcakmak Differential Revision: D61737790 --- ax/core/experiment.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/ax/core/experiment.py b/ax/core/experiment.py index 3cd8a16d5b3..860f6a5850d 100644 --- a/ax/core/experiment.py +++ b/ax/core/experiment.py @@ -1499,15 +1499,6 @@ def attach_trial( # Create the trial and add arm(s) trial = None if is_batch: - # TODO: HSS support for batch trials. - if self.search_space.is_hierarchical: - raise NotImplementedError( - "Support for batch trials " - "in hierarchical search space coming soon. Let " - "the Ax developers know if you have a use " - "case for it." - ) - trial = self.new_batch_trial( ttl_seconds=ttl_seconds, optimize_for_power=optimize_for_power ).add_arms_and_weights(arms=arms)