Prabakar
Databricks Employee
Databricks Employee

@Yoni Madar​ This issue can occur when we are running the code with non-ML cluster, though you install ML libraries. Using ML cluster will resolve this.

View solution in original post