Ajay-Pandey
Databricks MVP

Hi @Suky Muliadikara​ ,

The error is OutofMemory that means when you're calling display it's actually executing all the transformation that's causing OOM exception please increase your memory for driver and worker then you can use display ().

Ajay Kumar Pandey