Update neural_ontology_manager.py

dice-group · Dec 9, 2024 · 1a524f2 · 1a524f2
1 parent 8e135e9
commit 1a524f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/owlapy/owl_neural_reasoners/neural_ontology_manager.py b/owlapy/owl_neural_reasoners/neural_ontology_manager.py
@@ -1,12 +1,12 @@
-from owlapy.owl_ontology_manager import OntologyManager, Ontology
+from owlapy.owl_ontology_manager import OntologyManager
 from dicee.knowledge_graph_embeddings import KGE
 
 # LF: for now functions same as normal OntologyManager (+ can return path)
 class NeuralOntologyManager(OntologyManager):
 	def __init__(self, world_store=None):
 		super().__init__(world_store) 
 
-	def load_ontology(self, path: str = None) -> Ontology:
+	def load_ontology(self, path: str = None):
 		self._path = path
 		return super().load_ontology(path)