Bound streamed metadata scans to configured split size

Author: Boris Peyriguere

complexity/generative/detection/data.py

@@ -252,7 +252,7 @@ def _stream(self, *, metadata_only: bool = False):
         if metadata_only:
             stream = stream.cast_column(self.image_column, HFImage(decode=False))
             stream = stream.remove_columns(self.image_column)
-            return stream
+            return stream.take(self.total_examples)
         stream = stream.shuffle(
             seed=self.seed + self.epoch,
             buffer_size=self.shuffle_buffer,