transformers/tests/models/whisper
Matt f8b2574416
Better TF docstring types (#23477)
* Rework TF type hints to use | None instead of Optional[] for tf.Tensor

* Rework TF type hints to use | None instead of Optional[] for tf.Tensor

* Don't forget the imports

* Add the imports to tests too

* make fixup

* Refactor tests that depended on get_type_hints

* Better test refactor

* Fix an old hidden bug in the test_keras_fit input creation code

* Fix for the Deit tests
2023-05-24 13:52:52 +01:00
..
__init__.py Add WhisperModel to transformers (#19166) 2022-10-05 22:28:31 +02:00
test_feature_extraction_whisper.py is_batched fix for remaining 2-D numpy arrays (#23309) 2023-05-23 14:37:35 -04:00
test_modeling_flax_whisper.py Temporarily increase tol for PT-FLAX whisper tests (#23288) 2023-05-11 11:43:18 +01:00
test_modeling_tf_whisper.py Better TF docstring types (#23477) 2023-05-24 13:52:52 +01:00
test_modeling_whisper.py feat: Whisper prompting (#22496) 2023-05-19 09:33:11 +01:00
test_processor_whisper.py feat: Whisper prompting (#22496) 2023-05-19 09:33:11 +01:00
test_tokenization_whisper.py feat: Whisper prompting (#22496) 2023-05-19 09:33:11 +01:00