Skip to content

Commit 5c66698

Browse files
committed
refactor
1 parent 29add68 commit 5c66698

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

autointent/modules/scoring/_description/description.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -161,10 +161,7 @@ def predict(self, utterances: list[str]) -> NDArray[np.float64]:
161161
utterance_vectors = self._embedder.embed(utterances, TaskTypeEnum.sts)
162162
similarities: NDArray[np.float64] = cosine_similarity(utterance_vectors, self._description_vectors)
163163
else:
164-
pairs = []
165-
pairs.extend(
166-
[(utterance, description) for utterance in utterances for description in self._description_texts]
167-
)
164+
pairs = [(utterance, description) for utterance in utterances for description in self._description_texts]
168165

169166
scores = self._cross_encoder.predict(pairs)
170167
similarities = np.array(scores).reshape(len(utterances), len(self._description_texts))

0 commit comments

Comments
 (0)