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)