I don't know your setup but regular jupyterlab notebook execution is really exactly like python in almost everything, there's not much that should be different.