Fix the __getattr__ method in BatchEncoding (#4772)

This commit is contained in:
Julien Plu
2020-06-09 09:44:00 +02:00
committed by GitHub
parent 41a1d27cde
commit 9f5d5a531d
+4 -1
View File
@@ -266,7 +266,10 @@ class BatchEncoding(UserDict):
)
def __getattr__(self, item: str):
return self.data[item]
try:
return self.data[item]
except KeyError:
raise AttributeError
def keys(self):
return self.data.keys()