From c7722e1e8455ba43a67d1def8a381b0a6bba2a2e Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sol=C3=A8ne=20Tarride?= <starride@teklia.com>
Date: Wed, 27 Sep 2023 11:12:30 +0200
Subject: [PATCH] Improve code
---
dan/utils.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/dan/utils.py b/dan/utils.py
index fcd7e7af..c65df263 100644
--- a/dan/utils.py
+++ b/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):
--
GitLab