Update flashbax/buffers/item_buffer.py

Co-authored-by: Callum Tilbury <37700709+callumtilbury@users.noreply.github.com>

flashbax/buffers/item_buffer.py

@@ -109,7 +109,7 @@ def add_fn(
     def sample_fn(
         state: TrajectoryBufferState, rng_key: PRNGKey
     ) -> TrajectoryBufferSample[Experience]:
-        """Samples a batch of transitions from the buffer."""
+        """Samples a batch of items from the buffer."""
         sampled_batch = buffer.sample(state, rng_key).experience
         sampled_batch = jax.tree_map(lambda x: x.squeeze(axis=1), sampled_batch)
         return TrajectoryBufferSample(experience=sampled_batch)