transformers/tests/pipelines
Yusuf Shihata 9eac19eb59
[Feature] Support is_split_into_words in the TokenClassificationPipeline. (#38818)
* some fixes

* some fixes

* now the pipeline can take list of tokens as input and is_split_into_words argument

* now the pipeline can take list of tokens as input and is_split_into_words argument

* now the pipeline can take list of tokens as input and is_split_into_words argument and we can handle batches of tokenized input

* now the pipeline can take list of tokens as input and is_split_into_words argument and we can handle batches of tokenized input

* solving test problems

* some fixes

* some fixes

* modify tests

* aligning start and end correctly

* adding tests

* some formatting

* some formatting

* some fixes

* some fixes

* some fixes

* resolve conflicts

* removing unimportant lines

* removing unimportant lines

* generalize to other languages

* generalize to other languages

* generalize to other languages

* generalize to other languages
2025-06-23 15:31:32 +00:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_pipelines_audio_classification.py [CI] lazy loading external datasets (#37218) 2025-04-03 09:57:45 +01:00
test_pipelines_automatic_speech_recognition.py [whisper] move processor test into processor test file 🧹 (#38266) 2025-05-22 10:07:11 +01:00
test_pipelines_common.py Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
test_pipelines_depth_estimation.py [CI] lazy loading external datasets (#37218) 2025-04-03 09:57:45 +01:00
test_pipelines_document_question_answering.py 🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129) 2025-05-19 18:02:06 +01:00
test_pipelines_feature_extraction.py Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
test_pipelines_fill_mask.py Make pipeline able to load processor (#32514) 2024-10-09 16:46:11 +01:00
test_pipelines_image_classification.py [CI] lazy loading external datasets (#37218) 2025-04-03 09:57:45 +01:00
test_pipelines_image_feature_extraction.py Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
test_pipelines_image_segmentation.py Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
test_pipelines_image_text_to_text.py 🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129) 2025-05-19 18:02:06 +01:00
test_pipelines_image_to_image.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_image_to_text.py 🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129) 2025-05-19 18:02:06 +01:00
test_pipelines_mask_generation.py Expectation fixes and added AMD expectations (#38729) 2025-06-13 16:14:58 +02:00
test_pipelines_object_detection.py [CI] lazy loading external datasets (#37218) 2025-04-03 09:57:45 +01:00
test_pipelines_question_answering.py [BugFix] QA pipeline edge case: align_to_words=True in QuestionAnsweringPipeline can lead to duplicate answers (#38761) 2025-06-16 15:01:22 +00:00
test_pipelines_summarization.py 🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129) 2025-05-19 18:02:06 +01:00
test_pipelines_table_question_answering.py 🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129) 2025-05-19 18:02:06 +01:00
test_pipelines_text_classification.py Simplify soft dependencies and update the dummy-creation process (#36827) 2025-04-11 11:08:36 +02:00
test_pipelines_text_generation.py 🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129) 2025-05-19 18:02:06 +01:00
test_pipelines_text_to_audio.py 🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288) 2025-05-23 17:17:38 +02:00
test_pipelines_text2text_generation.py 🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129) 2025-05-19 18:02:06 +01:00
test_pipelines_token_classification.py [Feature] Support is_split_into_words in the TokenClassificationPipeline. (#38818) 2025-06-23 15:31:32 +00:00
test_pipelines_translation.py 🚨🚨🚨 [pipelines] update defaults in pipelines that can generate (#38129) 2025-05-19 18:02:06 +01:00
test_pipelines_video_classification.py [CI] lazy loading external datasets (#37218) 2025-04-03 09:57:45 +01:00
test_pipelines_visual_question_answering.py Use Python 3.9 syntax in tests (#37343) 2025-04-08 14:12:08 +02:00
test_pipelines_zero_shot_audio_classification.py Allow FP16 or other precision inference for Pipelines (#31342) 2024-07-05 17:21:50 +01:00
test_pipelines_zero_shot_image_classification.py Image pipelines spec compliance (#33899) 2024-10-08 13:34:28 +01:00
test_pipelines_zero_shot_object_detection.py Add image text to text pipeline (#34170) 2024-10-31 15:48:11 -04:00
test_pipelines_zero_shot.py Simplify soft dependencies and update the dummy-creation process (#36827) 2025-04-11 11:08:36 +02:00