move kg to tdc/utils folder

mims-harvard · Oct 30, 2023 · 5eee5a6 · 5eee5a6
1 parent fee45dd
commit 5eee5a6
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/tdc/__init__.py b/tdc/__init__.py
@@ -2,4 +2,4 @@
 from .oracles import Oracle
 from .benchmark_deprecated import BenchmarkGroup
 from .tdc_hf import tdc_hf_interface
-from .knowledge_graph import KnowledgeGraph
+from tdc.utils.knowledge_graph import KnowledgeGraph
diff --git a/tdc/resource/primekg.py b/tdc/resource/primekg.py
@@ -9,7 +9,7 @@
 import warnings
 
 from ..utils import general_load
-from ..knowledge_graph import KnowledgeGraph
+from ..utils.knowledge_graph import KnowledgeGraph
 
 warnings.filterwarnings("ignore")
 

diff --git a/tdc/knowledge_graph.py → tdc/utils/knowledge_graph.py b/tdc/knowledge_graph.py → tdc/utils/knowledge_graph.py
@@ -1,9 +1,7 @@
 """A python module to build, handle, explore, and manipulate knowledge graphs.
-This is mimicking __PrimeKG__'s pandas dataframe format
 """
 
 import pandas as pd
-
 from copy import copy
 
 kg_columns = [