Databricks Sql Partition By Case