diff --git a/tools/tpch.py b/tools/tpch.py index a89415c258..26025ad863 100644 --- a/tools/tpch.py +++ b/tools/tpch.py @@ -65,7 +65,7 @@ def run( parser.add_argument("--num-partitions", type=int, required=True, help="The number of partitions") parser.add_argument( "--cluster-profile", - choices=["debug_xs-x86", "medium-x86"], + choices=["debug_xs-x86", "medium-x86", "benchmarking-arm"], type=str, required=False, help="The ray cluster configuration to run on",