Merge pull request #93 from monarch-initiative/support_basemodel

Add support for BaseModel entities
monarch-initiative · Jul 16, 2022 · 79eabe3 · 79eabe3
2 parents 3eb5f9b + 0230a73
commit 79eabe3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/koza/__init__.py b/koza/__init__.py
@@ -1,2 +1,2 @@
 """Koza, an ETL framework for LinkML data models"""
-__version__ = '0.2.0'
+__version__ = '0.2.1'
diff --git a/koza/converter/kgx_converter.py b/koza/converter/kgx_converter.py
@@ -1,6 +1,6 @@
 from dataclasses import asdict
 from typing import Iterable, Tuple
-
+from pydantic import BaseModel
 
 class KGXConverter:
     """
@@ -40,7 +40,11 @@ def convert(self, entities: Iterable) -> Tuple[dict, dict]:
         return nodes, edges
 
     def convert_node(self, node) -> dict:
+        if isinstance(node, BaseModel):
+            return dict(node)
         return asdict(node)
 
     def convert_association(self, association) -> dict:
+        if isinstance(association, BaseModel):
+            return dict(association)
         return asdict(association)