Import Error using Azure DataBricks

[sennettleung]

Hi there,

I am trying to use ax-client to find the best hyperparameters for my neural network model, and I encountered this import error as shown in the link below. I used to be able to access the package okay, but it's not working now. I have installed all the dependency libraries onto the cluster, so am unsure why I'm still running into this error, using the pip install bleeding edge version of the package also didn't help resolve the issue. Any help is much appreciated, thank you!

Screenshot of the dependencies I have installed:

Screenshot of the error I encountered whilst trying to import the package (both lines output the same error message):

[saitcakmak]

Hi @sennettleung! I assume you're using gpytorch 1.9.0 here? That is currently not supported in Ax/BoTorch (will be supported in next release). If you downgrade to GPyTorch 1.8.1 things should work again.

[Balandat]

Related botorch issue: pytorch/botorch#1370

I am planning to release a version early next week to to fix this.

[sennettleung]

Hi both,

That's great, downgrading seemed to have worked, thank you for the update and workaround! :)

[lena-kashtelyan]

We're looking into cutting new version now, which should address this.

[saitcakmak]

This should be resolved in the just-released 0.2.7 version.