Hi,
I'm running example notebook from https://docs.databricks.com/aws/en/machine-learning/automl/regression-train-api on a node with ML cluster 17.0 (includes Apache Spark 4.0.0, Scala 2.13) and getting error at
from databricks import automl
summary = automl.regress(dataset=train_pdf, target_col="MedHouseVal", timeout_minutes=30)
TypeError: Retry.__init__() got an unexpected keyword argument 'method_whitelist'
File <command-910054366217001>, line 2 1 from databricks import automl ----> 2 summary = automl.regress(dataset=train_pdf, target_col="MedHouseVal", timeout_minutes=30)
Any ideas/recommendations why and how to fix it?
Regards,
Stas
