pip install is outdated?

the pip3 install ax-platform seems to be outdated. e.g. in ax/utils/tutorials/cnn_utils.py has different train and load_mnist functions compared to whats there on this github

def train(
    train_loader: DataLoader,
    parameters: Dict[str, float],
    dtype: torch.dtype,
    device: torch.device,
) -> nn.Module:

will throw error *** TypeError: train() got an unexpected keyword argument 'net' if I follow the tutorial on https://ax.dev/versions/latest/tutorials/tune_cnn.html net = train(net=net, train_loader=train_loader, parameters=parameterization, dtype=dtype, device=device)

Similarly train_loader, valid_loader, test_loader = load_mnist(batch_size=BATCH_SIZE) will also throw error

[Balandat]

The pip version is always the latest release and not meant to be up to date with the github master. You can find the tutorial for the current pip version here: https://ax.dev/versions/0.1.2/tutorials/tune_cnn.html

If you want to live on the bleeding edge and run the tutorial off master, you need to install Ax from the github master.

[kkashin]

Website versioning is fixed now, closing out the issue.