Не удается импортировать sklearn_pandas в блокнот Google AI Platform после установки pip

Я играю с блокнотом Google AI Platform Notebook (который на данный момент все еще находится в стадии бета-тестирования). Я загрузил скрипт Python, который зависит от sklearn_pandas, и я получаю эту ошибку:

---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
<ipython-input-4-24f79569b871> in <module>
----> 1 from sklearn_pandas import DataFrameMapper

ImportError: No module named 'sklearn_pandas'

Затем я пытаюсь установить его:

!pip install sklearn-pandas

Я перезапустил ядро, но оно все еще получает ту же ошибку ImportError. Я сделал это, чтобы подтвердить, что sklearn pandas действительно установлен:

!pip list |grep -i sklearn

DEPRECATION: The default format will switch to columns in the future. You can use --format=(legacy|columns) (or define a format=(legacy|columns) in your pip.conf under the [list] section) to disable this warning.
sklearn-pandas (1.8.0)

Кто-нибудь пробовал это? Я использовал google colab до этого, и у меня никогда не было таких трудностей. Было бы здорово, если бы уровень пользовательского опыта Google Colab можно было перенести в этот бета-продукт. Я думаю, что это может быть общая проблема с импортом модулей Python, а не только sklearn_pandas, в частности.