transformers/tests/utils
Matt 3bd1fe4315
Stop storing references to bound methods via tf.function (#24146)
* Stop storing references to bound methods in tf.functions

* Remove the gc.collect calls now that we resolved the underlying problem

* Remove the default signature from model.serving entirely, big cleanup

* Remove _prune_signature as self.input_signature can prune itself

* Restore serving docstring

* Update int support test to check the input signature

* Make sure other tests also use model.input_signature and not serving.input_signature

* Restore _prune_signature

* Remove the doctest GC now it's no longer needed

* Correct core tests to use the pruned sig

* order lines correctly in core tests

* Add eager_serving back with a deprecation warning
2023-06-13 19:04:22 +01:00
..
__init__.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_activations_tf.py TF: Add sigmoid activation function (#16819) 2022-04-19 16:13:08 +01:00
test_activations.py Add the GeLU activation from pytorch with the tanh approximation (#21345) 2023-02-02 09:33:04 -05:00
test_add_new_model_like.py Avoid invalid escape sequences, use raw strings (#22936) 2023-04-25 09:17:56 -04:00
test_audio_utils.py audio_utils improvements (#21998) 2023-05-09 09:10:17 -04:00
test_backbone_utils.py Add methods to update and verify out_features out_indices (#23031) 2023-05-04 10:15:06 +01:00
test_cli.py add trust_remote_code option to CLI download cmd (#24097) 2023-06-08 11:13:57 -04:00
test_convert_slow_tokenizer.py Revert error back into warning for byte fallback conversion. (#22607) 2023-04-06 14:00:29 +02:00
test_doc_samples.py [Test refactor 1/5] Per-folder tests reorganization (#15725) 2022-02-23 15:46:28 -05:00
test_dynamic_module_utils.py Fix the regex in get_imports to support multiline try blocks and excepts with specific exception types (#23725) 2023-05-24 15:40:19 -04:00
test_file_utils.py Inheritance-based framework detection (#21784) 2023-02-27 15:31:55 +00:00
test_generic.py Refactor conversion function (#19799) 2022-10-24 13:48:40 -04:00
test_hf_argparser.py Support union types X | Y syntax for HfArgumentParser for Python 3.10+ (#23126) 2023-05-03 10:49:54 -04:00
test_hub_utils.py Prepare tests for hfh 0.14 (#22958) 2023-04-24 09:31:50 -04:00
test_image_processing_utils.py Add Image Processors (#19796) 2022-11-02 11:57:36 +00:00
test_image_utils.py Accept batched tensor of images as input to image processor (#21144) 2023-01-26 10:15:26 +00:00
test_logging.py Fix flaky test for log level (#21776) 2023-02-28 16:24:14 -05:00
test_model_card.py Black preview (#17217) 2022-05-12 16:25:55 -04:00
test_model_output.py Fix ModelOutput instantiation when there is only one tuple (#20416) 2022-11-23 15:09:21 -05:00
test_modeling_tf_core.py Stop storing references to bound methods via tf.function (#24146) 2023-06-13 19:04:22 +01:00
test_offline.py Fix the expected error in test_offline_mode_pipeline_exception (#23022) 2023-04-27 14:22:05 +02:00
test_skip_decorators.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
test_versions_utils.py Update quality tooling for formatting (#21480) 2023-02-06 18:10:56 -05:00
tiny_model_summary.json Update tiny models and pipeline tests (#23446) 2023-05-18 17:29:04 +02:00