diff --git a/tests/generate/test_integration_llamacpp.py b/tests/generate/test_integration_llamacpp.py index 090a18ced..65db8fe1a 100644 --- a/tests/generate/test_integration_llamacpp.py +++ b/tests/generate/test_integration_llamacpp.py @@ -338,7 +338,7 @@ def function(foo: int, bar: List[int]): rng = torch.Generator(device="cpu") rng.manual_seed(10) - sequence = generate.json(model, function)( + sequence = generate.json(model, function, whitespace_pattern="")( prompt, max_tokens=100, temperature=0.0, rng=rng ) assert isinstance(sequence, dict)