From 49c4cf6a4c9b189f42c8076a6782d29011bd1ab7 Mon Sep 17 00:00:00 2001 From: Patrick Westphal Date: Fri, 21 Jun 2024 19:20:54 +0200 Subject: [PATCH] Updates on semanticlabeling.labelinferencer --- semanticlabeling/labelinferencer.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/semanticlabeling/labelinferencer.py b/semanticlabeling/labelinferencer.py index b4201ad..73343fe 100644 --- a/semanticlabeling/labelinferencer.py +++ b/semanticlabeling/labelinferencer.py @@ -5,12 +5,12 @@ 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 @@ -18,4 +18,8 @@ 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()