Skip to content

Commit eabe814

Browse files
committed
try to fix bug
1 parent 1de28f3 commit eabe814

File tree

2 files changed

+4
-8
lines changed

2 files changed

+4
-8
lines changed

autointent/modules/embedding/_logreg.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,8 @@ def from_context(
8181

8282
def clear_cache(self) -> None:
8383
"""Clear embedder from memory."""
84-
self._embedder.clear_ram()
84+
if hasattr(self, "_embedder"):
85+
self._embedder.clear_ram()
8586

8687
def fit(self, utterances: list[str], labels: ListOfLabels) -> None:
8788
"""Train the logistic regression model using the provided utterances and labels.
@@ -90,9 +91,6 @@ def fit(self, utterances: list[str], labels: ListOfLabels) -> None:
9091
utterances: List of text data to index
9192
labels: List of corresponding labels for the utterances
9293
"""
93-
if hasattr(self, "_embedder"):
94-
self.clear_cache()
95-
9694
self._validate_task(labels)
9795

9896
self._embedder = Embedder(

autointent/modules/embedding/_retrieval.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -83,9 +83,6 @@ def fit(self, utterances: list[str], labels: ListOfLabels) -> None:
8383
utterances: List of text data to index
8484
labels: List of corresponding labels for the utterances
8585
"""
86-
if hasattr(self, "_vector_index"):
87-
self.clear_cache()
88-
8986
self._validate_task(labels)
9087

9188
self._vector_index = VectorIndex(
@@ -140,7 +137,8 @@ def get_assets(self) -> EmbeddingArtifact:
140137

141138
def clear_cache(self) -> None:
142139
"""Clear cached data in memory used by the vector index."""
143-
self._vector_index.clear_ram()
140+
if hasattr(self, "_vector_index"):
141+
self._vector_index.clear_ram()
144142

145143
def predict(self, utterances: list[str]) -> list[ListOfLabels]:
146144
"""Predict the nearest neighbors for a list of utterances.

0 commit comments

Comments
 (0)