Models deprecated in Ax – issue in Assignments 4, 5, and 6

[Amanichab]

While working on Assignment 4, the following error appeared:

DeprecationWarning: Models is deprecated, use `ax.modelbridge.registry.Generators` instead.

The same issue also occurred in Assignments 5 and 6.

This is due to recent updates in the ax-platform library. The recommended approach is to use Generators instead.

rom ax.modelbridge.factory import Generators
from ax.generation_strategy.generation_strategy import GenerationStep, GenerationStrategy
gs = GenerationStrategy(
    steps=[
        GenerationStep(
            model=Generators.BOTORCH_MODULAR,
            num_trials=-1,
            max_parallelism=3,
        ),
    ]
)

[sgbaird]

Oh, right - thanks! I need to make sure requirements.txt pins to ax-platform<1. Could you make a PR for these into the public repos? Then, I need to merge that into the intermediate private template.