Ipython usage raises AttributeError

https://github.com/apache/beam/blob/master/sdks/python/apache_beam/runners/interactive/display/display_manager.py#L36

import IPython

_display_progress = IPython.display.display

This doesn't work:

In [1]: import IPython; IPython.display.display('test')
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-1-1ca36be5baf7> in <module>()
----> 1 import IPython; IPython.display.display('test')

AttributeError: 'module' object has no attribute 'display'
In [2]: from IPython import display; display.display('test')
'test'
In [3]: import IPython; IPython.display.display('test')
'test'

Should be:

import IPython

from IPython import display

_display_progress = display.display