Databricks interview question

Coding assignment. Signed NDA so cannot share.