Fix "spacy_model" key error when saving customize lemmas(#141)

xxyzz · Aug 6, 2023 · 9125d60 · 9125d60
1 parent fda940d
commit 9125d60
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/__init__.py b/__init__.py
@@ -2,7 +2,7 @@
 
 from calibre.customize import InterfaceActionBase
 
-VERSION = (3, 29, 5)
+VERSION = (3, 29, 6)
 
 
 class WordDumbDumb(InterfaceActionBase):

diff --git a/dump_lemmas.py b/dump_lemmas.py
@@ -66,7 +66,9 @@ def dump_spacy_docs(
     save_spacy_docs(
         nlp,
         spacy_model,
-        pkg_versions["spacy_model"],
+        pkg_versions[
+            "spacy_trf_model" if spacy_model.endswith("_trf") else "spacy_cpu_model"
+        ],
         lemma_lang,
         is_kindle,
         lemmas_conn,