transformers/tests/generation
Xiaoli Wang 239ace152b
Fix TypeError: Object of type int64 is not JSON serializable (#24340)
* Fix TypeError: Object of type int64 is not JSON serializable

* Convert numpy.float64 and numpy.int64 to float and int for json serialization

* Black reformatted examples/pytorch/token-classification/run_ner_no_trainer.py

* * make style
2023-06-27 12:15:49 +01:00
..
__init__.py
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 Time to Say Goodbye, torch 1.7 and 1.8 (#22291) 2023-03-21 19:22:01 +01:00
test_configuration_utils.py Generate: GenerationConfig can overwrite attributes at from_pretrained time (#24238) 2023-06-13 17:59:21 +01:00
test_flax_logits_process.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_flax_utils.py Add support for beam search's num_return_sequencs flag in flax (#23082) 2023-05-03 10:50:34 -04:00
test_framework_agnostic.py Fix TypeError: Object of type int64 is not JSON serializable (#24340) 2023-06-27 12:15:49 +01:00
test_logits_process.py Generate: add SequenceBiasLogitsProcessor (#24334) 2023-06-21 11:14:41 +01:00
test_stopping_criteria.py Generate: move generation_*.py src files into generation/*.py (#20096) 2022-11-09 15:34:08 +00:00
test_streamers.py Generate: TextIteratorStreamer timeout (#22576) 2023-04-05 09:57:46 +01:00
test_tf_logits_process.py Better TF docstring types (#23477) 2023-05-24 13:52:52 +01:00
test_tf_utils.py Better TF docstring types (#23477) 2023-05-24 13:52:52 +01:00
test_utils.py Generate: group_beam_search requires diversity_penalty>0.0 (#24456) 2023-06-27 10:46:39 +01:00