Update outlines/generate/api.py

Co-authored-by: Patrice Bechard <[email protected]>

Author: 2 people

outlines/generate/api.py

@@ -232,7 +232,7 @@ def __call__(
         # We reshape the output to (batch_size, sample_size)
         output: List[List[FormattedOutput]] = list()
         for i in range(0, batch_size * num_samples, num_samples):
-            output.append(formatted[i : i + num_samples]
+            output.append(formatted[i : i + num_samples])
 
         # We remove leading dimensions for the output
         if batch_size == 1 and num_samples == 1: