From 73094f3b2f35d26ee1ee3e80189bb022110528aa Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Sol=C3=A8ne=20Tarride?= <starride@teklia.com>
Date: Wed, 20 Sep 2023 14:17:11 +0200
Subject: [PATCH] Update tests for data extraction

---
 dan/datasets/extract/extract.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dan/datasets/extract/extract.py b/dan/datasets/extract/extract.py
index 0c1d2d13..74c7b327 100644
--- a/dan/datasets/extract/extract.py
+++ b/dan/datasets/extract/extract.py
@@ -280,7 +280,7 @@ class ArkindexExtractor:
         """
         return " ".join(
             [
-                self.mapping.encode[token] if token in self.mapping else token
+                self.mapping.encode[token] if token in self.mapping.encode else token
                 for token in list(text.strip())
             ]
         )
-- 
GitLab