SOM　somcoluについてのエラー文

somcoluについてのエラーメッセージ
以下のコードで自己組織化マップを用いた分類がしたいのですが、somocluというライブラリを使用すると以下のエラーが出ます。
NameError: name 'wrap_train' is not defined

importもしてあるのですが、何が問題かわかりません。
プログラミング初学者のため、簡単なところで間違っているかもしれませんがご教授お願いします。

python
1
2# coding: UTF-8
3import numpy as np
4from somoclu import Somoclu
5from sklearn.datasets import load_iris
6from sklearn.decomposition import PCA
7
8def main():
9    # データを読み込む
10    dataset = load_iris()
11    X = dataset.data
12    y = dataset.target
13
14    # SOMに入れる前にPCAして計算コスト削減を測る
15    pca = PCA(n_components=0.95)
16    X = pca.fit_transform(X)
17
18    # SOMの定義
19    n_rows = 16
20    n_cols = 24
21    som = Somoclu(n_rows=n_rows, n_columns=n_cols,initialization="pca", verbose=2)
22
23    # 学習
24    som.train(data=X, epochs=1000)
25
26    # U-matrixをファイル出力
27    som.view_umatrix(labels=y, bestmatches=True,
28                     filename="umatrix.png")
29
30
31if __name__ == "__main__":
32    main()