From 113706f67a4fbb66c515370983054dc0da4f1186 Mon Sep 17 00:00:00 2001 From: Samin Yasar Date: Wed, 24 May 2023 22:50:11 +0600 Subject: [PATCH 1/3] add type hint in pipeline model argument --- src/transformers/pipelines/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/pipelines/__init__.py b/src/transformers/pipelines/__init__.py index 84d461cd1ae730..963ce49385002a 100755 --- a/src/transformers/pipelines/__init__.py +++ b/src/transformers/pipelines/__init__.py @@ -505,7 +505,7 @@ def clean_custom_task(task_info): def pipeline( task: str = None, - model: Optional = None, + model: Optional[Union[str, Path]] = None, config: Optional[Union[str, PretrainedConfig]] = None, tokenizer: Optional[Union[str, PreTrainedTokenizer, "PreTrainedTokenizerFast"]] = None, feature_extractor: Optional[Union[str, PreTrainedFeatureExtractor]] = None, From 354431a269e11e132007d503b202131f7614f4ca Mon Sep 17 00:00:00 2001 From: Samin Yasar Date: Wed, 24 May 2023 23:35:27 +0600 Subject: [PATCH 2/3] add pretrainedmodel and tfpretainedmodel type hint --- src/transformers/pipelines/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/pipelines/__init__.py b/src/transformers/pipelines/__init__.py index 963ce49385002a..a497191535777e 100755 --- a/src/transformers/pipelines/__init__.py +++ b/src/transformers/pipelines/__init__.py @@ -505,7 +505,7 @@ def clean_custom_task(task_info): def pipeline( task: str = None, - model: Optional[Union[str, Path]] = None, + model: Optional[Union[str, PreTrainedModel, TFPreTrainedModel]] = None, config: Optional[Union[str, PretrainedConfig]] = None, tokenizer: Optional[Union[str, PreTrainedTokenizer, "PreTrainedTokenizerFast"]] = None, feature_extractor: Optional[Union[str, PreTrainedFeatureExtractor]] = None, From 070a32c0e5c4a2f035c0c2184d1356281050144c Mon Sep 17 00:00:00 2001 From: Samin Yasar Date: Wed, 24 May 2023 23:52:44 +0600 Subject: [PATCH 3/3] make type hints string --- src/transformers/pipelines/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/transformers/pipelines/__init__.py b/src/transformers/pipelines/__init__.py index a497191535777e..818164f3c28b6d 100755 --- a/src/transformers/pipelines/__init__.py +++ b/src/transformers/pipelines/__init__.py @@ -505,7 +505,7 @@ def clean_custom_task(task_info): def pipeline( task: str = None, - model: Optional[Union[str, PreTrainedModel, TFPreTrainedModel]] = None, + model: Optional[Union[str, "PreTrainedModel", "TFPreTrainedModel"]] = None, config: Optional[Union[str, PretrainedConfig]] = None, tokenizer: Optional[Union[str, PreTrainedTokenizer, "PreTrainedTokenizerFast"]] = None, feature_extractor: Optional[Union[str, PreTrainedFeatureExtractor]] = None,