XGBoostEstimator object not found in 5.0 ML Beta Runtime

The second line causes a runtime error in Databricks using the 5.0 ML Beta runtime configuration under Spark 2.4. The 4.1 ML configuration under Spark 2.3 works. Version 3 Python. I don't see reference to this bug - or if it is one identified.