Merge pull request stanfordnlp#1105 from tom-doerr/patch-16

okhat · web-flow · commit 747edc19fbe9 · 2024-06-11T10:39:46.000-07:00
Add message for assert
diff --git a/dspy/predict/multi_chain_comparison.py b/dspy/predict/multi_chain_comparison.py
@@ -42,7 +42,7 @@ def forward(self, completions, **kwargs):
                 f"«I'm trying to {rationale} I'm not sure but my prediction is {answer}»",
             )
 
-        assert len(attempts) == self.M, len(attempts)
+        assert len(attempts) == self.M, f"The number of attempts ({len(attempts)}) doesn't match the expected number M ({self.M}). Please set the correct value for M when initializing MultiChainComparison."
 
         kwargs = {
             **{

Original file line number	Diff line number	Diff line change
`@@ -42,7 +42,7 @@ def forward(self, completions, **kwargs):`
`42`	`42`	`f"«I'm trying to {rationale} I'm not sure but my prediction is {answer}»",`
`43`	`43`	`)`
`44`	`44`
`45`		`- assert len(attempts) == self.M, len(attempts)`
	`45`	`+ assert len(attempts) == self.M, f"The number of attempts ({len(attempts)}) doesn't match the expected number M ({self.M}). Please set the correct value for M when initializing MultiChainComparison."`
`46`	`46`
`47`	`47`	`kwargs = {`
`48`	`48`	`**{`