Skip to content

Commit

Permalink
Document what IterBlock does in code
Browse files Browse the repository at this point in the history
Signed-off-by: Ben Browning <[email protected]>
  • Loading branch information
bbrowning committed Dec 10, 2024
1 parent b980e75 commit b9100d6
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions src/instructlab/sdg/blocks/iterblock.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,21 @@
# This is part of the public API.
@BlockRegistry.register("IterBlock")
class IterBlock(Block):
"""
Call another block multiple times for a single set of input
samples, concatening the results of each iteration's call to that
other block in the final returned output.
Args:
num_iters: The number of times to iterate over the block
block_type: The type of the other block to call (ie LLMBlock)
block_config: Any necessary configuration that will get passed to the
other block to properly configure it.
Returns:
A Dataset containing all output samples from each iteration
"""

def __init__(
self,
ctx,
Expand Down

0 comments on commit b9100d6

Please sign in to comment.