アルゴリズムとかオーダーとか

仕事で勉強したことなどをまとめてます

Failed to import transformers.trainer because of the following error の解決方法

huggingfaceのtransformersを使っていたら、急に「Failed to import transformers.trainer because of the following error」というエラーが発生した。対応方法がなかなか見つからなかったので記事に残しておく。

環境

Windows 11 Home + WSL2 + Ubuntu-20.04

conda version : 22.9.0
conda-build version : 3.21.9
python version : 3.8.13.final.0

ライブラリのバージョン

pytorch-transformers 1.2.0 pypi_0 pypi
transformers 4.18.0 py38h06a4308_0
pytorch 1.12.1 py3.8_cuda11.6_cudnn8.3.2_0 pytorch
pytorch-memlab 0.2.4 pypi_0 pypi
pytorch-mutex 1.0 cuda pytorch
pytorch-transformers 1.2.0 pypi_0 pypi
torchaudio 0.12.1 py38_cu116 pytorch
torchvision 0.13.1 py38_cu116 pytorch

Stack trace

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
File ~/anaconda3/lib/python3.8/site-packages/transformers/utils/import_utils.py:1002, in _LazyModule._get_module(self, module_name)
   1001 try:
-> 1002     return importlib.import_module("." + module_name, self.__name__)
   1003 except Exception as e:

File ~/anaconda3/lib/python3.8/importlib/__init__.py:127, in import_module(name, package)
    126         level += 1
--> 127 return _bootstrap._gcd_import(name[level:], package, level)

File <frozen importlib._bootstrap>:1014, in _gcd_import(name, package, level)

File <frozen importlib._bootstrap>:991, in _find_and_load(name, import_)

File <frozen importlib._bootstrap>:975, in _find_and_load_unlocked(name, import_)

File <frozen importlib._bootstrap>:671, in _load_unlocked(spec)

File <frozen importlib._bootstrap_external>:843, in exec_module(self, module)

File <frozen importlib._bootstrap>:219, in _call_with_frames_removed(f, *args, **kwds)

File ~/anaconda3/lib/python3.8/site-packages/transformers/trainer.py:176
    175 if is_datasets_available():
--> 176     import datasets
    178 if is_torch_tpu_available(check_device=False):

File ~/anaconda3/lib/python3.8/site-packages/datasets/__init__.py:43
     41 del version
---> 43 from .arrow_dataset import Dataset
     44 from .arrow_reader import ReadInstruction

File ~/anaconda3/lib/python3.8/site-packages/datasets/arrow_dataset.py:62
     60 from tqdm.auto import tqdm
---> 62 from . import config
     63 from .arrow_reader import ArrowReader

File ~/anaconda3/lib/python3.8/site-packages/datasets/config.py:41
     40 # Imports
---> 41 DILL_VERSION = version.parse(importlib_metadata.version("dill"))
     42 PANDAS_VERSION = version.parse(importlib_metadata.version("pandas"))

File ~/anaconda3/lib/python3.8/site-packages/packaging/version.py:49, in parse(version)
     48 try:
---> 49     return Version(version)
     50 except InvalidVersion:

File ~/anaconda3/lib/python3.8/site-packages/packaging/version.py:264, in Version.__init__(self, version)
    261 def __init__(self, version: str) -> None:
    262 
    263     # Validate the version and parse it into pieces
--> 264     match = self._regex.search(version)
    265     if not match:

TypeError: expected string or bytes-like object

The above exception was the direct cause of the following exception:

RuntimeError                              Traceback (most recent call last)
Cell In [27], line 3
      1 #@title Trainerの定義
----> 3 from transformers import Trainer
      4 from transformers import TrainingArguments
      5 from transformers import EarlyStoppingCallback

File <frozen importlib._bootstrap>:1039, in _handle_fromlist(module, fromlist, import_, recursive)

File ~/anaconda3/lib/python3.8/site-packages/transformers/utils/import_utils.py:992, in _LazyModule.__getattr__(self, name)
    990     value = self._get_module(name)
    991 elif name in self._class_to_module.keys():
--> 992     module = self._get_module(self._class_to_module[name])
    993     value = getattr(module, name)
    994 else:

File ~/anaconda3/lib/python3.8/site-packages/transformers/utils/import_utils.py:1004, in _LazyModule._get_module(self, module_name)
   1002     return importlib.import_module("." + module_name, self.__name__)
   1003 except Exception as e:
-> 1004     raise RuntimeError(
   1005         f"Failed to import {self.__name__}.{module_name} because of the following error (look up to see its"
   1006         f" traceback):\n{e}"
   1007     ) from e

RuntimeError: Failed to import transformers.trainer because of the following error (look up to see its traceback):
expected string or bytes-like object

エラー原因

以下の組み合わせだと今回のエラーが発生する。
datasets==2.6.1
dill==0.3.5.1

エラー発生個所はdatasetsの以下の部分
https://github.com/huggingface/datasets/blob/a2576b8c1664806e814d2427bf21bbef4be85456/src/datasets/config.py#L41

エラー原因はどうやらdillのバージョンがimportlib.metadata.versionで取得できないからのようだ。実際にコンソールで試してみても取得できない。

Python 3.8.13 (default, Mar 28 2022, 11:38:47)
[GCC 7.5.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import importlib.metadata
>>> importlib.metadata.version("dill")
>>>

dill==0.3.6だとversion番号が変えてくるので、dill=0.3.5.1が悪さをしている模様。なので、0.3.6にしたいところだが、実はdatasets=2.6.1がまだdill==0.3.6に対応していないため、setup.pyにdill<0.3.6の指定がある。このため、強制的にdill==0.3.5.1にダウングレードされてしまう。

解決策

dill==0.3.5をインストールしたらよいかと思ったが、どうもconda installでインストールできなかった。multiprocessが依存ライブラリとしてdill >= 0.3.6を指定しており、コンフリクトが起きるためなのかな?

最終的には、datasets=2.4.0にすることで解決できた。

なお、google colabでリモート接続の場合だとdill=0.3.5.1でも問題なく動いているので、おま環の可能性が高い。