Skip to content

Commit

Permalink
dict copy 😭
Browse files Browse the repository at this point in the history
  • Loading branch information
varun-magesh committed Jun 14, 2024
1 parent b45d160 commit 26ecad7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion rl/llm/openai_batched.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,7 @@ def prepare_batch(self) -> list[dict]:
}
batch = []
for n, req in enumerate(self.request):
formatted_request = batch_template.copy()
formatted_request = dict(batch_template)
formatted_request["custom_id"] = f"{self.id_prefix}{n}"
formatted_request["body"]["messages"] = req
batch.append(formatted_request)
Expand Down

0 comments on commit 26ecad7

Please sign in to comment.