Lol, OK so in my case it was because I had a file called databricks.py with clashed with the installed databricks. Renaming my file to databricks_utils.py solved it.
Did anyone make any progress here? I seem to have the same issue. It works in an interactive shell, but doesn't work in my code. File "/home/ubuntu/change-detection-inference/liveeo/flows/change_detection_inference/flow.py", line 8, in <module>
...
