fix length based example selector (#862)

pull/868/head
Harrison Chase 2 years 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 = [] examples = []
while remaining_length > 0 and i < len(self.examples): while remaining_length > 0 and i < len(self.examples):
new_length = remaining_length - self.example_text_lengths[i] new_length = remaining_length - self.example_text_lengths[i]
if i < 0: if new_length < 0:
break break
else: else:
examples.append(self.examples[i]) examples.append(self.examples[i])

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

Loading…
Cancel
Save