Hi @DE_Saheed, thanks for reaching out! Welcome to the GX community.
We had a recent related thread that explored how to get the GX context to write to a DBFS-mounted folder, I think this will have the examples you need to help get your context working on Databricks.
Could you take a look and see if that helps solve your issue?