Databricks: «Numba требует NumPy 1.22» при обновлении numpy

Я использую Azure Databricks с Python. Помимо прочего, я использую модули umap и numpy. Когда я пытаюсь import umap.umap_ as umap, я получаю эту ошибку:

Ошибка импорта: Numba требуется NumPy 1.22 или более поздней версии. Получил NumPy 1.21.

Хорошо, просто обновите numpy, верно? Я побежал !{sys.executable} -m pip install numpy --upgrade и это сработало. Numpy теперь показывает версию в проводнике модуля:

Но когда я снова пытаюсь импортировать umap, я получаю ту же ошибку, что и раньше, говоря, что требуется версия 1.22+. Это требование соблюдено, хотя версия 1.26 похоже не загружается. Я попытался загрузить правильную версию, используя это:

import pkg_resources
pkg_resources.require("numpy==1.26.0")
import numpy as np

Все еще не повезло. Это дает:

VersionConflict: (numpy 1.21.5 (/databricks/python3/lib/python3.9/site-packages), Requirement.parse('numpy==1.26.0'))

Так как мне это сделать?