Fix mypy errors of pyoxigraph v0.4.0 API change

xxyzz · Sep 23, 2024 · 8e50f9f · 8e50f9f
1 parent bb3710a
commit 8e50f9f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/proficiency/extract_dbnary.py b/src/proficiency/extract_dbnary.py
@@ -3,7 +3,7 @@
 from pathlib import Path
 from shutil import which
 
-from pyoxigraph import Store
+from pyoxigraph import RdfFormat, Store
 
 from .database import create_indexes_then_close, init_db, wiktionary_db_path
 from .util import (
@@ -238,13 +238,13 @@ def insert_senses(
 def init_oxigraph_store(gloss_lang: str) -> tuple[Store, bool]:
     gloss_lang = convert_lang_code(gloss_lang)
     store = Store(f"build/ttl/{gloss_lang}_store")
-    store.bulk_load(f"build/ttl/{gloss_lang}_dbnary_ontolex.ttl", "text/turtle")
+    store.bulk_load(f"build/ttl/{gloss_lang}_dbnary_ontolex.ttl", RdfFormat.TURTLE)
     exolex_path = Path(f"build/ttl/{gloss_lang}_dbnary_exolex_ontolex.ttl")
     if exolex_path.exists():
-        store.bulk_load(str(exolex_path), "text/turtle")
+        store.bulk_load(str(exolex_path), RdfFormat.TURTLE)
     morphology_path = Path(f"build/ttl/{gloss_lang}_dbnary_morphology.ttl")
     if morphology_path.exists():
-        store.bulk_load(str(morphology_path), "text/turtle")
+        store.bulk_load(str(morphology_path), RdfFormat.TURTLE)
     store.optimize()
 
     return store, morphology_path.exists()
@@ -292,7 +292,7 @@ def convert_lang_code(code: str) -> str:
     gloss_lang = "fr"
     store = Store()
     # store.bulk_load(f"ttl/test.ttl", "text/turtle")
-    store.bulk_load(f"build/ttl/{gloss_lang}_dbnary_ontolex.ttl", "text/turtle")
+    store.bulk_load(f"build/ttl/{gloss_lang}_dbnary_ontolex.ttl", RdfFormat.TURTLE)
     # store.bulk_load(f"ttl/{gloss_lang}_dbnary_morphology.ttl", "text/turtle")
     # store.bulk_load(f"ttl/{gloss_lang}_dbnary_exolex_ontolex.ttl", "text/turtle")
     store.optimize()