fixed imports

dice-group · Dec 9, 2024 · 4045219 · 4045219
1 parent 1a524f2
commit 4045219
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/owlapy/owl_neural_reasoners/neural_ontology_manager.py b/owlapy/owl_neural_reasoners/neural_ontology_manager.py
@@ -1,16 +1,16 @@
-from owlapy.owl_ontology_manager import OntologyManager
+from owlapy.owl_ontology_manager import OntologyManager, Ontology
 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):
+	def load_ontology(self, path: str = None) -> Ontology:
 		self._path = path
 		return super().load_ontology(path)
 
-	def load_neural_embedding(self, path: str = None):
+	def load_neural_embedding(self, path: str = None) -> KGE:
 		self._path_neural_embedding = path	
 		return KGE(path)
 

diff --git a/owlapy/owl_neural_reasoners/owl_neural_reasoner.py b/owlapy/owl_neural_reasoners/owl_neural_reasoner.py
@@ -14,7 +14,6 @@
 from collections import Counter, OrderedDict
 from functools import lru_cache
 from owlapy.owl_neural_reasoners.abstract import AbstractNeuralReasoner
-from owlapy.owl_ontology import Ontology
 from owlapy.owl_neural_reasoners.neural_ontology_manager import NeuralOntologyManager
 
 # TODO: