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

[g-simmons]

trlx/trlx/pipeline/offline_pipeline.py

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]],

[LouisCastricato]

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