"s is not iterable" on PySpark Kernel

Checklist prior to opening an issue

• I have followed fully the installation steps laid out in the documentation site.
• I have restarted jupyterlab.
• I have read the FAQ section in the documentation site.

Describe the bug I successfully installed jupyterlab_code_formatter and it works on Python notebooks; however, I would also like it to format PySpark notebooks (still Python), but I get the following error: s is not iterable. The workaround is to switch the kernel back to Python then format then rever it back to PySpark. Is there a way to fix this via the config?

I’m assuming this line of code is relevant

Diagnostic commands Please attach the output of the following commands (please format them properly)

• pip freeze

• jupyter labextension list

• jupyter serverextension list

Screenshots If applicable, add screenshots to help explain your problem.

Additional context Add any other context about the problem here.

Happy to provide the diagnostic command if that’s helpful here! Also happy to contribute.

[ryantam626]

Yep looks sensible, a PR would be greatly appreciated - did you test with a local installation of the updated plugin by the way?

[devstein]

@ryantam626 Works! Thanks for all the help on this 🥇