Skip to content

Commit

Permalink
Fixed type hint for pipelines/check_task (huggingface#19150)
Browse files Browse the repository at this point in the history
  • Loading branch information
Fei-Wang authored Sep 23, 2022
1 parent ece7624 commit 6395d12
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/transformers/pipelines/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -385,7 +385,7 @@ def get_task(model: str, use_auth_token: Optional[str] = None) -> str:
return task


def check_task(task: str) -> Tuple[Dict, Any]:
def check_task(task: str) -> Tuple[str, Dict, Any]:
"""
Checks an incoming task string, to validate it's correct and return the default Pipeline and Model classes, and
default models if they exist.
Expand Down

0 comments on commit 6395d12

Please sign in to comment.