Make sure IME state is updated on BubbleExpandedView

This was only getting set in certain situations; make sure
to always set it.

Flag: EXEMPT bugfix
Test: manual - expand bubbles on phone landscape, trigger IME
             - rotate device so it's portrait
             => observe that expanded view is at the right size
Bug: 397135369
Change-Id: Id92d2028446ff754284ae23b979e90d9eef38f0b