Updates on semanticlabeling.labelinferencer

semantic-systems · Jun 21, 2024 · 49c4cf6 · 49c4cf6
1 parent 97f0ecd
commit 49c4cf6
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/semanticlabeling/labelinferencer.py b/semanticlabeling/labelinferencer.py
@@ -5,17 +5,21 @@
 
 import util.graphbuilder
 from semanticlabeling.labeledcolumn import LabeledColumn
-from util.ontology import Ontology
+from util.knowledgesource import KnowledgeSource
 from util.file import InputFile
 
 
 class SemanticLabelInferencer(ABC):
-    def __init__(self, input_file: InputFile, ontologies: List[Ontology]):
+    def __init__(self, input_file: InputFile, ontologies: List[KnowledgeSource]):
         self.input_file = input_file
         self.ontologies = ontologies
 
     def get_labeled_columns(self) -> List[LabeledColumn]:
         return self.input_file.columns
 
     def get_graph(self) -> BankGraph:
-        return util.graphbuilder.build(self.input_file)
+        return util.graphbuilder.build(self.input_file.columns)
+
+    def print_summary(self):
+        pass  # TODO
+        raise NotImplementedError()