Skip to content

Commit fb48ed9

Browse files
authored
Update test_outline.py
1 parent 0bd1547 commit fb48ed9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_outline.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,9 @@ def template(a: int) -> str:
1616

1717
def test_outline():
1818
mock_model = Mock()
19-
mock_model.__iter__ = lambda self: iter([])
19+
mock_model.vocabulary.__iter__ = lambda self: iter([])
2020
mock_generator = Mock()
21-
mock_generator.__iter__ = lambda self: iter([])
21+
mock_generator.vocabulary.__iter__ = lambda self: iter([])
2222
mock_generator.return_value = '{"result": 6}'
2323
with patch("outlines.generate.json", return_value=mock_generator):
2424
outline_instance = Outline(mock_model, template, OutputModel)

0 commit comments

Comments
 (0)