Mlflow tracing not working

[LiamNiisan]

Describe the bug
When trying to use the tracing features of mlflow in AzureML:

https://mlflow.org/docs/latest/tracing/

There is an error on the client side:

mlflow.tracking.client: Failed to start trace AsyncCompletions: API request to endpoint /api/2.0/mlflow/traces failed with error code 403 != 200. Response body

To Reproduce
Steps to reproduce the behavior:

1. You just need to setup mlflow as usual with the right tracking uri.
2. Add the line: mlflow.openai.autolog()

Expected behavior
It shouldn't crash the code as autolog is quite standard, even if tracing isn't supported, it should at the very least save the traces somewhere.

[github-actions]

Thanks for the feedback! We are routing this to the appropriate team for follow-up. cc @Azure/azure-ml-sdk @azureml-github.

[github-actions]

Hello @LiamNiisan. I'm an AI assistant for the azure-sdk-for-python repository. I have some suggestions that you can try out while the team gets back to you.

• The error (HTTP 403) suggests the tracking endpoint for traces may be rejecting the request—this might be a permissions or feature support issue on the Azure ML side.

• Depending on your use case, consider wrapping the autolog call in a try/except so that failures in starting trace reporting do not crash your workflow.

• It may be necessary to disable or bypass tracing if it isn’t supported by the configured MLFlow backend. Refer to the MLFlow tracing documentation for guidance on configuring tracing and understanding supported scenarios: MLFlow Tracing.

• If tracing is critical to your application, please file a GitHub issue with additional logs and context so that we can further diagnose if the AzureML tracing integration can be adjusted.

• As a temporary workaround, review your tracking URI configuration and consider if a fallback (or explicit disable) of autolog tracing for AzureML can be implemented.

The team will get back to you shortly, hopefully this helps in the meantime.

[kingernupur]

Thank you for reporting the issue @LiamNiisan!
Please note that mlflow tracing is not supported in AML yet.