Update test_outline.py

dottxt-ai · Jan 28, 2025 · 0bd1547 · 0bd1547
1 parent cc33df1
commit 0bd1547
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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)