Fix the __getattr__ method in BatchEncoding (#4772)

Fix the getattr method in BatchEncoding (#4772)
9f5d5a53 · Julien Plu · GitHub · 41a1d27c · 9f5d5a53
Unverified Commit 9f5d5a53 authored Jun 09, 2020 by Julien Plu Committed by GitHub Jun 09, 2020
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

src/transformers/tokenization_utils.py src/transformers/tokenization_utils.py +4 -1

No files found.
--- a/src/transformers/tokenization_utils.py
+++ b/src/transformers/tokenization_utils.py
@@ -266,7 +266,10 @@ class BatchEncoding(UserDict):
            )

    def __getattr__(self, item: str):
+        try:
            return self.data[item]
+        except KeyError:
+            raise AttributeError

    def keys(self):
        return self.data.keys()