Choice parameters fail

I am trying out Ax with a simple set of ‘choice’ parameters. Here is my script

from ax import optimize
best_parameters, best_values, experiment, model = optimize(
        parameters=[
          {
            "name": "x1",
            "type": "choice",
            "values": [-10, 0, 10],
          },
          {
            "name": "x2",
            "type": "choice",
            "values": [-10, 0, 10],
          },
        ],
        # Booth function
        evaluation_function=lambda p: (p["x1"] + 2*p["x2"] - 7)**2 + (2*p["x1"] + p["x2"] - 5)**2,
        minimize=True,
    )

During the 6th trial, I am getting this error TypeError: optimize_acqf_discrete() got an unexpected keyword argument 'num_restarts'

Looks like mk_ehvi_default_optimizer_options() in ax\models\torch\botorch_modular\default_options.py is setting this argument. I am not sure why this error is occurring only in the 6th trial and not before that.

Following is my environment:

1. Python 3.8.11
2. ax-platform==0.2.1
3. botorch==0.5.0
4. Windows 10

[lena-kashtelyan]

We are working on the new release, it should hopefully be out within the next two days.

[imbalu007]

No problem. Please update when the fix is released and I will give it another try. Thanks!