Skip to content

Commit

Permalink
default batch_size to 5
Browse files Browse the repository at this point in the history
  • Loading branch information
parkervg committed Oct 16, 2024
1 parent 3b69d3b commit b481118
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions blendsql/ingredients/builtin/map/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -119,7 +119,7 @@ def from_args(
model: Optional[Model] = None,
few_shot_examples: Optional[List[dict]] = None,
list_options_in_prompt: bool = True,
batch_size: Optional[int] = None,
batch_size: Optional[int] = 5,
k: Optional[int] = None,
):
"""Creates a partial class with predefined arguments.
Expand All @@ -129,7 +129,7 @@ def from_args(
few_shot_examples: A list of dictionary MapExample few-shot examples.
If not specified, will use [default_examples.json](https://github.com/parkervg/blendsql/blob/main/blendsql/ingredients/builtin/map/default_examples.json) as default.
list_options_in_prompt: Whether to list options in the prompt. Defaults to True.
batch_size: The batch size for processing. Defaults to None.
batch_size: The batch size for processing. Defaults to 5.
k: Determines number of few-shot examples to use for each ingredient call.
Default is None, which will use all few-shot examples on all calls.
If specified, will initialize a haystack-based embedding retriever to filter examples.
Expand Down

0 comments on commit b481118

Please sign in to comment.