transformers/tests/generation
Arthur 44e3e3fb49
prepare for "__floordiv__ is deprecated and its behavior will change in a future version of pytorch" (#20211)
* rounding_mode = "floor"  instead of // to prevent behavioral change

* add other TODO

* use `torch_int_div` from pytrch_utils

* same for tests

* fix copies

* style

* use relative imports when needed

* Co-authored-by: sgugger <sylvain.gugger@gmail.com>
2023-03-01 10:49:21 +01:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_beam_constraints.py Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_beam_search.py prepare for "__floordiv__ is deprecated and its behavior will change in a future version of pytorch" (#20211) 2023-03-01 10:49:21 +01:00
test_configuration_utils.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_flax_logits_process.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_flax_utils.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_framework_agnostic.py Generate: TF supports multiple eos tokens (#21571) 2023-02-13 12:24:22 +00:00
test_logits_process.py Add hallucination filter (#18675) 2023-01-19 11:20:25 -05:00
test_stopping_criteria.py Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_tf_logits_process.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_tf_utils.py Generate: filter encoder inputs when its signature does not accept wildcards (#21603) 2023-02-14 10:46:46 +00:00
test_utils.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00