The customer is trying to execute the Python code with kind Spark. They are able to execute the same code in Databricks so we suggested executing only the query in SQL Component, but they were trying the entire code in SQL Component which we used in the code component. 

Solution:-

We connected for a quick meeting and asked them to execute the Python code with the kind Python. and it worked fine.