Aha! Downgrading to pyviz_comms==2.3.2 fixed the problem for me. Thanks @ahuang11 !
pyviz_comms==2.3.2