riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Fixed type hint for pipelines/check_task (#19150)

Browse Source
This commit is contained in:
Fei Wang 2022-09-24 02:35:19 +08:00 committed by GitHub
parent ece762443e
commit 6395d1227f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/pipelines/__init__.py
View File

@ -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.