Skip to content

Commit b94653a

Browse files
committed
feat: change predict in RetrievalEmbedding
1 parent 7c4a5e4 commit b94653a

File tree

1 file changed

+0
-6
lines changed

1 file changed

+0
-6
lines changed

autointent/modules/embedding/_retrieval.py

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -302,12 +302,6 @@ class RetrievalEmbedding(EmbeddingModule):
302302
db_dir=db_dir,
303303
)
304304
retrieval.fit(utterances, labels)
305-
predictions = retrieval.predict(["how is the weather today?"])
306-
print(predictions)
307-
308-
.. testoutput::
309-
310-
([[1, 1]], [[0.1525942087173462, 0.18616724014282227]], [['good morning', 'how are you?']])
311305
312306
.. testcleanup::
313307

0 commit comments

Comments
 (0)