Noopur_Nigam
Databricks Employee
Databricks Employee

Could you please provide your full code? I would also like to know which DBR version you are using in your cluster.