TrajectoryPlanning problem

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,

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"),
 }