Skip to content

Commit

Permalink
Apply isort and black reformatting
Browse files Browse the repository at this point in the history
Signed-off-by: tbartley94 <tbartley94@users.noreply.github.com>
  • Loading branch information
tbartley94 committed Jul 22, 2024
1 parent e1e9b92 commit 6b4cb7e
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion nemo/collections/common/prompts/canary.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,9 @@ class CanaryPromptFormatter(PromptFormatter):
"template": f"{CANARY_BOS}|source_lang||task||target_lang||pnc|",
"slots": {
"source_lang": Modality.Text,
"task": Modality.TextLiteral("asr", "ast", "translate", "transcribe", "s2t_translation", "<|transcribe|>", "<|translate|>"),
"task": Modality.TextLiteral(
"asr", "ast", "translate", "transcribe", "s2t_translation", "<|transcribe|>", "<|translate|>"
),
"target_lang": Modality.Text,
"pnc": Modality.TextLiteral("yes", "no", "<|pnc|>", "<|nopnc|>"),
},
Expand Down

0 comments on commit 6b4cb7e

Please sign in to comment.