• Matt's avatar
    TF/Numpy variants for all DataCollator classes (#13105) · 854260ca
    Matt authored
    
    
    * Adding a TF variant of the DataCollatorForTokenClassification to get feedback
    
    * Added a Numpy variant and a post_init check to fail early if a missing import is found
    
    * Fixed call to Numpy variant
    
    * Added a couple more of the collators
    
    * Update src/transformers/data/data_collator.py
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    
    * Fixes, style pass, finished DataCollatorForSeqToSeq
    
    * Added all the LanguageModeling DataCollators, except SOP and PermutationLanguageModeling
    
    * Adding DataCollatorForPermutationLanguageModeling
    
    * Style pass
    
    * Add missing `__call__` for PLM
    
    * Remove `post_init` checks for frameworks because the imports inside them were making us fail code quality checks
    
    * Remove unused imports
    
    * First attempt at some TF tests
    
    * A second attempt to make any of those tests actually work
    
    * TF tests, round three
    
    * TF tests, round four
    
    * TF tests, round five
    
    * TF tests, all enabled!
    
    * Style pass
    
    * Merging tests into `test_data_collator.py`
    
    * Merging tests into `test_data_collator.py`
    
    * Fixing up test imports
    
    * Fixing up test imports
    
    * Trying shuffling the conditionals around
    
    * Commenting out non-functional old tests
    
    * Completed all tests for all three frameworks
    
    * Style pass
    
    * Fixed test typo
    
    * Style pass
    
    * Move standard `__call__` method to mixin
    
    * Rearranged imports for `test_data_collator`
    
    * Fix data collator typo "torch" -> "pt"
    
    * Fixed the most embarrassingly obvious bug
    
    * Update src/transformers/data/data_collator.py
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    
    * Renaming mixin
    
    * Updating docs
    Co-authored-by: default avatarSylvain Gugger <35901082+sgugger@users.noreply.github.com>
    Co-authored-by: default avatarDalton Walker <dalton_walker@icloud.com>
    Co-authored-by: default avatarAndrew Romans <andrew.romans@hotmail.com>
    854260ca
test_data_collator.py 39.2 KB