Set cell height in Colab

from IPython.core.magic import register_line_magic
from IPython.display import Javascript

@register_line_magic
def set_cell_height(size=300):
  display(
      Javascript("google.colab.output.setIframeHeight(0, true, {maxHeight: " +
                 str(size) + "})"))

# ==== In cell ====
%set_cell_height 300