Skip to content

Commit 11b78e3

Browse files
author
Yuwei Yan
committed
fix(tests): add missing contributions argument to paragraph planning tests
The _plan_flat_paragraphs() and _plan_subsection_paragraphs() methods were updated in a previous commit to require a contributions parameter, but the test calls were not updated accordingly.
1 parent cbc35ba commit 11b78e3

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

tests/test_incremental_planning_models.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -552,6 +552,7 @@ async def mock_llm(system, user, label, **kwargs):
552552
word_budget=800,
553553
reference_keys=["ref1", "ref2"],
554554
prior_key_points="",
555+
contributions=[],
555556
)
556557
assert len(paragraphs) == 3
557558
assert all(isinstance(p, ParagraphPlan) for p in paragraphs)
@@ -580,6 +581,7 @@ async def mock_llm(system, user, label, **kwargs):
580581
word_budget=1000,
581582
reference_keys=["ref1"],
582583
prior_key_points="introduction covered background and gap",
584+
contributions=[],
583585
)
584586

585587
prompt = captured_prompts[0]
@@ -674,6 +676,7 @@ async def mock_llm(system, user, label, **kwargs):
674676
section=section,
675677
subsection_structure=subsection_structure,
676678
reference_keys=["ref1", "ref2"],
679+
contributions=[],
677680
)
678681

679682
assert len(result) == 3
@@ -722,6 +725,7 @@ async def mock_llm(system, user, label, **kwargs):
722725
section=section,
723726
subsection_structure=subsection_structure,
724727
reference_keys=[],
728+
contributions=[],
725729
)
726730

727731
assert "Transitioning from A to B" in captured_prompts[1]
@@ -762,6 +766,7 @@ async def mock_llm(system, user, label, **kwargs):
762766
section=section,
763767
subsection_structure=subsection_structure,
764768
reference_keys=[],
769+
contributions=[],
765770
)
766771

767772
assert result[0].mission == "Evaluate on VQA benchmarks."

0 commit comments

Comments
 (0)