Databricks interview question

Coding. Just general coding question