llmblock: fix batched=False

_generate() returns a list of samples. The previouis code would create
a list of lists when batching was turned off. This restores that case
back to a list of samples.

Signed-off-by: Russell Bryant <[email protected]>

src/instructlab/sdg/llmblock.py

@@ -118,7 +118,7 @@ def generate(self, samples, **gen_kwargs) -> Dataset:
         if batched:
             outputs = self._generate(samples, **gen_kwargs)
         else:
-            outputs = [self._generate([sample], **gen_kwargs) for sample in samples]
+            outputs = [self._generate([sample], **gen_kwargs)[0] for sample in samples]
 
         new_data = []
         for sample, output in zip(samples, outputs):