jupyter notebook でmglearnがimportできない

ジュピターノートブックでmglearnをインポートしようとするとモジュールエラーになってしまいます。
インストールはしていてパスもあっていると思うのですが

PS C:\Users\shota\Documents> python
Python 3.7.4 (default, Aug  9 2019, 18:34:13) [MSC v.1915 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
>>> import mglearn
>>> mglearn.__file__
'C:\Users\shota\Anaconda3\lib\site-packages\mglearn\__init__.py'
>>> import pip
>>> pip.__file__
'C:\Users\shota\Anaconda3\lib\site-packages\pip\__init__.py'

動作環境
python 3.8
windows
powershell
anaconda3

実行したいコード

from sklearn.linear_model import LinearRegression

X,y = mglearn.datasets.make_wave(n_samples = 60)
X_train,X_test,y_train,y_test = train_test_split(X,yrandom_state = 42 )
lr = LinearRegression().fit(X_train,y_train)

追記

jupyternotebook エラー

NameError                                 Traceback (most recent call last)
<ipython-input-1-4900cee0be20> in <module>
      1 from sklearn.linear_model import LinearRegression
      2 
----> 3 X,y = mglearn.datasets.make_wave(n_samples = 60)
      4 X_train,X_test,y_train,y_test = train_test_split(X,yrandom_state = 42 )
      5 lr = LinearRegression().fit(X_train,y_train)

NameError: name 'mglearn' is not defined