Thanks for reaching out to us, are you using SDK v1 for your development?
If you are using SDK v2, please check on the samples below for your reference -
https://github.com/Azure/azureml-examples/tree/main/sdk/python
The package you are using in v2 for Azure Machine Learning should be -
pip install azure-ai-ml
Please have a try and let us know how it works! I hope this helps.
Regards,
Yutong
-Please kindly accept the answer if you feel helpful to support the community, thanks lot.