transformers/tests/trainer
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_data_collator.py handle numpy inputs in whole word mask data collator (#22032) 2023-03-10 10:50:29 -05:00
test_trainer.py Refactor hyperparameter search backends (#24384) 2023-06-22 14:28:25 -04:00
test_trainer_callback.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00
test_trainer_distributed.py Introduce PartialState as the device handler in the Trainer (#22752) 2023-04-17 15:09:45 -04:00
test_trainer_seq2seq.py Fix TypeError: Object of type int64 is not JSON serializable (#24340) 2023-06-27 12:15:49 +01:00
test_trainer_tpu.py
test_trainer_utils.py Apply ruff flake8-comprehensions (#21694) 2023-02-22 09:14:54 +01:00