undo

BlueBrain · Nov 28, 2023 · aeaa4d8 · aeaa4d8
1 parent cbdf767
commit aeaa4d8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/kgforge/core/forge.py b/kgforge/core/forge.py
@@ -45,6 +45,8 @@
 )
 from kgforge.core.reshaping import Reshaper
 from kgforge.core.wrappings.paths import PathsWrapper, wrap_paths, Filter
+from kgforge.specializations.mappers import DictionaryMapper
+from kgforge.specializations.mappings import DictionaryMapping
 
 
 class KnowledgeGraphForge:
@@ -547,7 +549,7 @@ def mappings(
 
     @catch
     def mapping(
-        self, entity: str, source: str, type: Type[Mapping] = "DictionaryMapping"
+        self, entity: str, source: str, type: Type[Mapping] = DictionaryMapping
     ) -> Mapping:
         """
         Return a Mapping object of type 'type' for a resource type 'entity' and a source.
@@ -564,7 +566,7 @@ def map(
         self,
         data: Any,
         mapping: Union[Mapping, List[Mapping]],
-        mapper: Type[Mapper] = "DictionaryMapper",
+        mapper: Type[Mapper] = DictionaryMapper,
         na: Union[Any, List[Any]] = None,
     ) -> Union[Resource, List[Resource]]:
         """