From 2a8b1979a68ed4e43125ee34b7d7720600041e70 Mon Sep 17 00:00:00 2001 From: David Eriksson Date: Fri, 13 Feb 2026 17:23:58 -0800 Subject: [PATCH] TrajectoryPlanning problem (#4908) Summary: Adds the new TrajectoryPlanning problem to the Ax. Reviewed By: hvarfner Differential Revision: D92568833 --- ax/benchmark/problems/registry.py | 10 ++++++++++ ax/benchmark/problems/synthetic/from_botorch.py | 1 + 2 files changed, 11 insertions(+) diff --git a/ax/benchmark/problems/registry.py b/ax/benchmark/problems/registry.py index 875e7c98782..238d0d327c0 100644 --- a/ax/benchmark/problems/registry.py +++ b/ax/benchmark/problems/registry.py @@ -25,6 +25,7 @@ from ax.benchmark.problems.synthetic.hss.jenatton import get_jenatton_benchmark_problem from botorch.test_functions import synthetic from botorch.test_functions.multi_objective import BraninCurrin +from botorch.test_functions.synthetic import TrajectoryPlanning @dataclass @@ -203,6 +204,15 @@ class BenchmarkProblemRegistryEntry: "observe_noise_sd": False, }, ), + "trajectory_planning_30": BenchmarkProblemRegistryEntry( + factory_fn=create_problem_from_botorch, + factory_kwargs={ + "test_problem_class": TrajectoryPlanning, + "test_problem_kwargs": {"dim": 30}, + "num_trials": 50, + "observe_noise_sd": False, + }, + ), # Problems where we observe the noise level "branin_observed_noise": BenchmarkProblemRegistryEntry( factory_fn=create_problem_from_botorch, diff --git a/ax/benchmark/problems/synthetic/from_botorch.py b/ax/benchmark/problems/synthetic/from_botorch.py index 697d7b7e0f0..56aa0294db2 100644 --- a/ax/benchmark/problems/synthetic/from_botorch.py +++ b/ax/benchmark/problems/synthetic/from_botorch.py @@ -62,6 +62,7 @@ ("StyblinskiTang", 2): -50.22278471716156, ("TensionCompressionString", None): float("inf"), ("ThreeHumpCamel", None): 3.7321680621434155, + ("TrajectoryPlanning", 30): 5.350095657055234, ("WeldedBeamSO", None): float("inf"), }