From 6cd34d451c8bcefb011907485fd6f5ee6828507f Mon Sep 17 00:00:00 2001 From: jiangmingyan <1829166702@qq.com> Date: Thu, 15 Jun 2023 19:33:37 +0800 Subject: [PATCH] [fix] bug in BatchEncoding.__getitem__ (#24293) Co-authored-by: luchen --- src/transformers/tokenization_utils_base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/tokenization_utils_base.py b/src/transformers/tokenization_utils_base.py index ba2f8ef3f1..eb757d98ac 100644 --- a/src/transformers/tokenization_utils_base.py +++ b/src/transformers/tokenization_utils_base.py @@ -242,7 +242,7 @@ class BatchEncoding(UserDict): elif self._encodings is not None: return self._encodings[item] elif isinstance(item, slice): - return {key: self.data[key][slice] for key in self.data.keys()} + return {key: self.data[key][item] for key in self.data.keys()} else: raise KeyError( "Invalid key. Only three types of key are available: "