Limit the number of inputs for a pipeline step

[malteos]

Initial Checks

• I have searched Google & GitHub for similar requests and couldn't find anything
• I have read and followed the documentation and still think this feature is missing

Description

For testing and debugging purposes, it would be great if we could limit/select the number of input samples when running a specific step.

Example:

# take the first N example
wurzel run steps.my_bugging_but_expensive_step --take_first 5

# take the specific examples by index
wurzel run steps.my_bugging_but_expensive_step --take_by_index 5,10,93