CarperAI / trlx

Lines 108 to 116 in 0dce99d

    
           @register_datapipeline 
        
           class PromptPipeline(BasePipeline): 
        
               """ 
        
               Dataloader which is used to supply prompts for either training or evaluation 
        
               Args: 
        
                   prompts (`List[str]` or `List[Dict[str, Any]]`): list of raw text prompts or a dictionary with a required 
        
                       key `"prompt"` and extra information, that would be passed along the generation for that prompt as a 
        
                       keyword argument to a reward function.

trlx/trlx/pipeline/offline_pipeline.py

Line 126 in 0dce99d

prompts: Union[Dict[str, Any], List[str]],

would be an easy PR if you wanna fix it ;)

	@register_datapipeline
	class PromptPipeline(BasePipeline):
	"""
	Dataloader which is used to supply prompts for either training or evaluation

	Args:
	prompts (`List[str]` or `List[Dict[str, Any]]`): list of raw text prompts or a dictionary with a required
	key `"prompt"` and extra information, that would be passed along the generation for that prompt as a
	keyword argument to a reward function.

Type hint doesn't match documentation for prompts arg of PromptPipeline.__init__

Type hint doesn't match documentation for prompts arg of PromptPipeline.init