Improve code

dan/utils.py

@@ -36,7 +36,7 @@ class LMTokenMapping(NamedTuple):
         return {a.display: a.encoded for a in self}
 
     def encode_token(self, token: str) -> str:
-        return self.encode[token] if token in self.encode else token
+        return self.encode.get(token, token)
 
 
 class EntityType(NamedTuple):