Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add Tests for extend generation logic #1184

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
Prev Previous commit
Next Next commit
feat(dspy): rm redundant comments
  • Loading branch information
JONEMI19 committed Jun 19, 2024
commit ea29442c7aea461e91e5b6d14edaafb1075e91f4
3 changes: 0 additions & 3 deletions tests/predict/test_predict.py
Original file line number Diff line number Diff line change
Expand Up @@ -242,7 +242,6 @@ class SandwichIdea(dspy.Signature):


def test_extend_generation(SandwichIdea):
# test that the completion is extended correctly
lm = DummyLM(
[
" whole wheat\n\nProtein: turkey\n\nFat: avocado",
Expand All @@ -260,7 +259,6 @@ def test_extend_generation(SandwichIdea):


def test_extend_generation_rolled_back_when_field_is_skipped(SandwichIdea):
# test the completion is rolled back when a field is skipped
lm = DummyLM(
[
" white\n\nFat: butter\n\nGarish: lettuce\n\nSauce: mayo",
Expand All @@ -278,7 +276,6 @@ def test_extend_generation_rolled_back_when_field_is_skipped(SandwichIdea):


def test_extend_generation_with_empty_field(SandwichIdea):
# test the completion is extended if the field is empty
lm = DummyLM(
[
" white\n\nProtein: \n\nFat: butter\n\nGarish: lettuce",
Expand Down
Loading