missing import azure.ai.ml.sweep in the sample code

[RohitMungi-MSFT]

Which example? Describe the issue

This notebook is missing the following import statement for the cell "Configure the Command" step 2.1

from azure.ai.ml.sweep import Choice, Uniform, MedianStoppingPolicy

Original issue reported from Azure docs
https://github.com/MicrosoftDocs/azure-docs/issues/98282

example: Add the import statement in step 2.1 for the cell "Configure the Command"
from azure.ai.ml.sweep import Choice, Uniform, MedianStoppingPolicy

description:
Without this import an error will be reported when the cell 3.1 "Modify or define inputs to sweep" is run.

Additional context