fix length based example selector (#862)

makefile-update-1
Harrison Chase 1 year ago committed by GitHub
parent 3f48eed5bd
commit 8df6b68093
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -51,7 +51,7 @@ class LengthBasedExampleSelector(BaseExampleSelector, BaseModel):
examples = []
while remaining_length > 0 and i < len(self.examples):
new_length = remaining_length - self.example_text_lengths[i]
if i < 0:
if new_length < 0:
break
else:
examples.append(self.examples[i])

@ -17,7 +17,7 @@ def selector() -> LengthBasedExampleSelector:
selector = LengthBasedExampleSelector(
examples=EXAMPLES,
example_prompt=prompts,
max_length=25,
max_length=30,
)
return selector

Loading…
Cancel
Save