Skip to content

Commit 5a32f44

Browse files
committed
fix: mypy
1 parent 1bbc0ba commit 5a32f44

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

autointent/modules/embedding/_retrieval.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -226,8 +226,8 @@ def get_assets(self) -> RetrieverArtifact:
226226
return RetrieverArtifact(embedder_name=self.embedder_name)
227227

228228
def clear_cache(self) -> None:
229-
"""Clear cached data in memory used by the embedder."""
230-
super().clear_cache()
229+
"""Clear cached data in memory used by the vector index."""
230+
self.vector_index.clear_ram()
231231

232232
def dump(self, path: str) -> None:
233233
"""
@@ -275,7 +275,7 @@ def load(self, path: str) -> None:
275275
self.label_encoder = LabelEncoder()
276276
self.label_encoder.classes_ = self.metadata["classes"]
277277

278-
def predict(self, utterances: list[str]) -> tuple[list[list[int | list[int]]], list[list[float]], list[list[str]]]:
278+
def predict(self, utterances: list[str]) -> None:
279279
pass
280280

281281

0 commit comments

Comments
 (0)