Type fix for generate enrichment graph (#44)

* Updated pandasaurus version to 0.3.7 * Refactored generate_enrichment_graph method
INCATools · Nov 3, 2023 · 02591f4 · 02591f4
1 parent e7865f2
commit 02591f4
Show file tree

Hide file tree

Showing 4 changed files with 250 additions and 196 deletions.
diff --git a/pandasaurus/graph/graph_generator.py b/pandasaurus/graph/graph_generator.py
@@ -2,7 +2,7 @@
 
 import networkx as nx
 import pandas as pd
-from rdflib import RDF, RDFS, Graph, Literal, Namespace, URIRef
+from rdflib import OWL, RDF, RDFS, Graph, Literal, Namespace, URIRef
 from rdflib.plugins.sparql import prepareQuery
 
 from pandasaurus.graph.graph_generator_utils import (
@@ -37,7 +37,9 @@ def generate_enrichment_graph(enriched_df: pd.DataFrame) -> Graph:
             s = cl_namespace[row["s"].split(":")[-1]]
             o = cl_namespace[row["o"].split(":")[-1]]
             graph.add((s, RDFS.label, Literal(row["s_label"])))
+            graph.add((s, RDF.type, OWL.Class))
             graph.add((o, RDFS.label, Literal(row["o_label"])))
+            graph.add((o, RDF.type, OWL.Class))
             graph.add((s, RDFS.subClassOf, o))
         return graph