riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 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 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 Checks an incoming task string, to validate it's correct and return the default Pipeline and Model classes, and
default models if they exist. default models if they exist.