r.inspect() exception with Release 2.0.1

[Brad Barker]

In Databricks 12.2LTS, when running r.inspect() within training/notebooks/Train get the following exception:
Uncaught TypeError: Cannot read properties of null (reading 'querySelectorAll')

When updating requirements.txt from mlflow==2.0.1 to mlflow==2.2.2 was able to get r.inspect() to render the pipeline graphic.

[Arpit Jasapara]

We updated the MLflow version requirement, thanks!