fix type hint in PromptPipeline.__init__

[g-simmons]

Fixes #494

[LouisCastricato]

Ah I assumed you'd change the doc string. Did you verify that this change doesn't break anything?

[g-simmons]

Oh, I assumed it was the type hint that was incorrect 😅.

Do you think the change is correct? Or is Dict[str, Any] actually one of the expected input types for prompts?

Would I need to do anything other than running the tests to confirm this doesn't break anything?

[g-simmons]

Why I thought this change was correct:

• eval_prompts is documented to accept List[Dict[str,Any]] in train fn of trlx.py
• it seems like the first element of prompts is expected to sometimes be a dict in PromptPipeline.__init__.

[cat-state]

LGTM! Thanks @g-simmons

[g-simmons]

🖖