UI does not render code correctly for python spark compute

To reproduce

1. Deploy dynamic spark-compute v2.0.6 from cask market
2. Use Python spark compute node
3. Place the following code in the text area for code

4. Reopen the Python spark compute node

The UI will render the code poorly formatted. Attached screenshots