From 0bd1547fc7707ab24f6bf4841ab25ed62eaf7c68 Mon Sep 17 00:00:00 2001 From: Yvan Sraka Date: Tue, 28 Jan 2025 19:40:54 +0100 Subject: [PATCH] Update test_outline.py --- tests/test_outline.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/test_outline.py b/tests/test_outline.py index 2c809ce31..9ffe31f69 100644 --- a/tests/test_outline.py +++ b/tests/test_outline.py @@ -18,6 +18,7 @@ def test_outline(): mock_model = Mock() mock_model.__iter__ = lambda self: iter([]) mock_generator = Mock() + mock_generator.__iter__ = lambda self: iter([]) mock_generator.return_value = '{"result": 6}' with patch("outlines.generate.json", return_value=mock_generator): outline_instance = Outline(mock_model, template, OutputModel)