Cleanup

biopragmatics · Apr 4, 2024 · f415ab0 · f415ab0
1 parent 397a65e
commit f415ab0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/src/semra/pipeline.py b/src/semra/pipeline.py
@@ -279,10 +279,10 @@ def get_raw_mappings(configuration: Configuration) -> t.List[Mapping]:
 def process(
     mappings: t.List[Mapping],
     upgrade_prefixes=None,
-    remove_prefix_set: t.Optional[t.Collection[str]]=None,
-    keep_prefix_set: t.Optional[t.Collection[str]]=None,
-    post_remove_prefixes: t.Optional[t.Collection[str]]=None,
-    post_keep_prefixes: t.Optional[t.Collection[str]]=None,
+    remove_prefix_set: t.Optional[t.Collection[str]] = None,
+    keep_prefix_set: t.Optional[t.Collection[str]] = None,
+    post_remove_prefixes: t.Optional[t.Collection[str]] = None,
+    post_keep_prefixes: t.Optional[t.Collection[str]] = None,
     *,
     remove_imprecise: bool = True,
 ) -> t.List[Mapping]:

diff --git a/src/semra/sources/wikidata.py b/src/semra/sources/wikidata.py
@@ -7,7 +7,10 @@
 from semra.rules import EXACT_MATCH, UNSPECIFIED_MAPPING
 from semra.struct import Mapping, MappingSet, SimpleEvidence
 
-__all__ = ["get_wikidata_mappings"]
+__all__ = [
+    "get_wikidata_mappings",
+    "get_wikidata_mappings_by_prefix",
+]
 
 
 def get_wikidata_mappings(prop: str, predicate: t.Optional[Reference] = None) -> t.List[Mapping]: