Skip to content

Commit b6a0c85

Browse files
authored
Update embedder prompt (#183)
1 parent 2c6ace8 commit b6a0c85

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

autointent/_embedder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -178,7 +178,7 @@ def embed(self, utterances: list[str], task_type: TaskTypeEnum | None = None) ->
178178
convert_to_numpy=True,
179179
batch_size=self.config.batch_size,
180180
normalize_embeddings=True,
181-
prompt_name=self.config.get_prompt_type(task_type),
181+
prompt=self.config.get_prompt_type(task_type),
182182
)
183183

184184
if self.config.use_cache:

0 commit comments

Comments
 (0)