colab_window.py

# Run this cell only once if you are having issues with your input/output...
# window overflowing and scrolling to the left and right horizontally.
# This code is a very hacky way of trying to adjust this.
# It resizes colab output window so it doesn't overflow
from IPython.display import HTML, display

def set_css():
  display(HTML('''
  <style>
    pre {
        white-space: pre-wrap;
    }
  </style>
  '''))
get_ipython().events.register('pre_run_cell', set_css)