Fix type hints.

Author: gouttegd

src/sssom/rdf_internal.py

@@ -9,7 +9,7 @@
 from curies import Converter
 from linkml_runtime.linkml_model.meta import SlotDefinition
 from linkml_runtime.utils.schemaview import SchemaView
-from pandas import DataFrame
+from pandas import DataFrame, Series
 from rdflib import BNode, Graph, Literal, Node, URIRef
 from rdflib.namespace import RDF, RDFS, XSD
 from typing_extensions import override
@@ -36,6 +36,7 @@
 __all__ = ["MappingSetRDFConverter"]
 
 TRIPLE = tuple[Node, Node, Node]
+DICT_OR_SERIES = Union[Dict[str, Any], Series]
 MAPPINGS_IRI = URIRef(MAPPINGS, SSSOM_URI_PREFIX)
 EXTENSION_DEFINITION_IRI = URIRef(EXTENSION_DEFINITIONS, SSSOM_URI_PREFIX)
 
@@ -563,7 +564,7 @@ def _multivalue_from_rdf(self, value: Any, slot_name: str, dest: Dict[str, Any])
     # Conversion to RDF
     #
 
-    def dict_to_rdf(self, g: Graph, obj: Dict[str, Any]) -> Node:
+    def dict_to_rdf(self, g: Graph, obj: DICT_OR_SERIES) -> Node:
         """Export a SSSOM object to a RDF graph.
 
         :param g: The graph to export the object to.
@@ -611,7 +612,7 @@ def _is_empty(self, value: Any) -> bool:
         """
         return value is None or (hasattr(value, "__len__") and len(value) == 0)
 
-    def _init_dict_to_rdf(self, g: Graph, obj: Dict[str, Any]) -> Node:
+    def _init_dict_to_rdf(self, g: Graph, obj: DICT_OR_SERIES) -> Node:
         """Create the root node representing a SSSOM object.
 
         Subclasses should override this method to customize the way
@@ -897,7 +898,7 @@ def msdf_to_rdf(
         return g
 
     @override
-    def _init_dict_to_rdf(self, g: Graph, obj: Dict[str, Any]) -> Node:
+    def _init_dict_to_rdf(self, g: Graph, obj: DICT_OR_SERIES) -> Node:
         if MAPPING_SET_ID in obj:
             return URIRef(obj[MAPPING_SET_ID])
         else:
@@ -926,7 +927,7 @@ def _process_slot(self, g: Graph, subject: Node, name: str, value: Any) -> bool:
         return done
 
     def _mapping_to_rdf(
-        self, g: Graph, subject: Node, mapping: Dict[str, Any], hydrate: bool
+        self, g: Graph, subject: Node, mapping: DICT_OR_SERIES, hydrate: bool
     ) -> None:
         mapping_node = self.mapping_converter.dict_to_rdf(g, mapping)
         g.add(cast(TRIPLE, [subject, MAPPINGS_IRI, mapping_node]))
@@ -979,7 +980,7 @@ def _multivalue_from_rdf(self, value: Any, slot_name: str, dest: Dict[str, Any])
             dest[slot_name] = dest[slot_name] + "|" + value
 
     @override
-    def _init_dict_to_rdf(self, g: Graph, obj: Dict[str, Any]) -> Node:
+    def _init_dict_to_rdf(self, g: Graph, obj: DICT_OR_SERIES) -> Node:
         if RECORD_ID in obj:
             return URIRef(self.ccp().expand(obj[RECORD_ID]))
         else:

tests/test_parsers.py

@@ -10,7 +10,7 @@
 from pathlib import Path
 from tempfile import TemporaryDirectory
 from textwrap import dedent
-from typing import Any, Dict, Union, get_args
+from typing import Any, Dict, Union, cast, get_args
 from xml.dom import minidom
 
 import numpy as np
@@ -329,13 +329,13 @@ def test_parse_standard_and_old_style_rdf(self) -> None:
             CONFIDENCE: 0.7,
             MAPPING_DATE: date(2025, 10, 27),
         }
-        assert_dict_contains(self, expected_values, msdf.df.loc[0])
+        assert_dict_contains(self, expected_values, cast(pd.Series, msdf.df.loc[0]))
 
         msdf = parse_sssom_rdf(f"{test_data_dir}/pre-standard-rdf.ttl")
         self.assertEqual(msdf.metadata[MAPPING_SET_ID], "https://example.org/sets/standard-rdf")
         self.assertEqual(msdf.metadata[LICENSE], "https://creativecommons.org/licenses/by/4.0/")
         self.assertEqual(msdf.metadata[PUBLICATION_DATE], date(2025, 10, 28))
-        assert_dict_contains(self, expected_values, msdf.df.loc[0])
+        assert_dict_contains(self, expected_values, cast(pd.Series, msdf.df.loc[0]))
 
     def test_parse_rdf_with_record_id(self) -> None:
         """Test parsing a RDF file containing record IDs as named resources."""