Update quality tooling for formatting (#21480)
* Result of black 23.1 * Update target to Python 3.7 * Switch flake8 to ruff * Configure isort * Configure isort * Apply isort with line limit * Put the right black version * adapt black in check copies * Fix copies
This commit is contained in:
@@ -430,7 +430,6 @@ class BigBirdModelTester:
|
||||
|
||||
@require_torch
|
||||
class BigBirdModelTest(ModelTesterMixin, unittest.TestCase):
|
||||
|
||||
# head masking & pruning is currently not supported for big bird
|
||||
test_head_masking = False
|
||||
test_pruning = False
|
||||
|
||||
@@ -24,6 +24,7 @@ from ...test_modeling_flax_common import FlaxModelTesterMixin, ids_tensor, rando
|
||||
|
||||
if is_flax_available():
|
||||
import jax
|
||||
|
||||
from transformers.models.big_bird.modeling_flax_big_bird import (
|
||||
FlaxBigBirdForCausalLM,
|
||||
FlaxBigBirdForMaskedLM,
|
||||
@@ -134,7 +135,6 @@ class FlaxBigBirdModelTester(unittest.TestCase):
|
||||
|
||||
@require_flax
|
||||
class FlaxBigBirdModelTest(FlaxModelTesterMixin, unittest.TestCase):
|
||||
|
||||
all_model_classes = (
|
||||
(
|
||||
FlaxBigBirdForCausalLM,
|
||||
@@ -210,7 +210,6 @@ class FlaxBigBirdModelTest(FlaxModelTesterMixin, unittest.TestCase):
|
||||
|
||||
self.assertEqual(len(outputs), len(jitted_outputs))
|
||||
for jitted_output, output in zip(jitted_outputs, outputs):
|
||||
|
||||
self.assertEqual(jitted_output.shape, output.shape)
|
||||
|
||||
# overwrite from common in order to skip the check on `attentions`
|
||||
|
||||
@@ -30,7 +30,6 @@ SAMPLE_VOCAB = get_tests_dir("fixtures/test_sentencepiece.model")
|
||||
@require_sentencepiece
|
||||
@require_tokenizers
|
||||
class BigBirdTokenizationTest(TokenizerTesterMixin, unittest.TestCase):
|
||||
|
||||
tokenizer_class = BigBirdTokenizer
|
||||
rust_tokenizer_class = BigBirdTokenizerFast
|
||||
test_rust_tokenizer = True
|
||||
|
||||
Reference in New Issue
Block a user