Skip to content

Commit 1767341

Browse files
committed
updates
1 parent 9a2ec46 commit 1767341

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/pipelines/test_pipelines_common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2043,8 +2043,8 @@ def test_encode_prompt_works_in_isolation(self):
20432043
encode_prompt_tree = ast_vistor.get_ast_tree(cls=self.pipeline_class)
20442044
ast_vistor.visit(encode_prompt_tree)
20452045
prompt_embed_kwargs = ast_vistor.return_names
2046-
print(f"{prompt_embed_kwargs=}")
20472046
prompt_embeds_kwargs = dict(zip(prompt_embed_kwargs, encoded_prompt_outputs))
2047+
20482048
# Pack the outputs of `encode_prompt`.
20492049
adapted_prompt_embeds_kwargs = {
20502050
k: prompt_embeds_kwargs.pop(k) for k in list(prompt_embeds_kwargs.keys()) if k in pipe_call_parameters

0 commit comments

Comments
 (0)