Change white space pattern in llama.cpp test

rlouf · rlouf · commit 121a25c62e46 · 2024-04-08T17:51:49.000+02:00
diff --git 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)

Original file line number	Diff line number	Diff line change
`@@ -338,7 +338,7 @@ def function(foo: int, bar: List[int]):`
`338`	`338`
`339`	`339`	`rng = torch.Generator(device="cpu")`
`340`	`340`	`rng.manual_seed(10)`
`341`		`- sequence = generate.json(model, function)(`
	`341`	`+ sequence = generate.json(model, function, whitespace_pattern="")(`
`342`	`342`	`prompt, max_tokens=100, temperature=0.0, rng=rng`
`343`	`343`	`)`
`344`	`344`	`assert isinstance(sequence, dict)`