ModuleNotFoundError: No module named 'botorch.models.fidelity'

Run a simple optimization loop throws ModuleNotFoundError, using ax-platform==0.1.6 version.

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

Traceback

Traceback (most recent call last): from ax.models.torch.botorch_defaults import ( File “/home/zoran/MyProjects/mindsdb-examples/pl/lib/python3.7/site-packages/ax/models/torch/botorch_defaults.py”, line 12, in <module> from botorch.models.fidelity.gp_regression_fidelity import ( ModuleNotFoundError: No module named ‘botorch.models.fidelity’

It looks like there are changes in the latest botorch 0.2.0 version. Downgrading to 0.1.4 fixes the issue.

[lena-kashtelyan]

This should be resolved in latest Ax, 0.1.8. Please reopen if the issue does not go away for you with version update, @ZoranPandovski!

[Balandat]

1. The botorch pip version is already 0.2.0, correct. You can use that with the Ax master - you can install the Ax master via pip using pip install git+ssh://[e-mail]/facebook/Ax.git as described in the readme.

2. You can install the specific botorch version from pip using pip install -e git://github.com/pytorch/botorch.git@v0.1.4. Then you can install the Ax release as usual.