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:
@@ -20,6 +20,7 @@ import unittest
|
||||
from huggingface_hub import HfFolder, delete_repo, set_access_token
|
||||
from parameterized import parameterized
|
||||
from requests.exceptions import HTTPError
|
||||
|
||||
from transformers import AutoConfig, GenerationConfig
|
||||
from transformers.testing_utils import TOKEN, USER, is_staging_test
|
||||
|
||||
|
||||
@@ -27,6 +27,7 @@ from ..test_modeling_flax_common import ids_tensor
|
||||
if is_flax_available():
|
||||
import jax
|
||||
import jax.numpy as jnp
|
||||
|
||||
from transformers.generation import (
|
||||
FlaxForcedBOSTokenLogitsProcessor,
|
||||
FlaxForcedEOSTokenLogitsProcessor,
|
||||
|
||||
@@ -27,6 +27,7 @@ if is_flax_available():
|
||||
|
||||
import jax.numpy as jnp
|
||||
from jax import jit
|
||||
|
||||
from transformers import AutoTokenizer, FlaxAutoModelForCausalLM
|
||||
from transformers.modeling_flax_pytorch_utils import load_flax_weights_in_pytorch_model
|
||||
|
||||
|
||||
@@ -8,7 +8,6 @@ from transformers import AutoTokenizer
|
||||
|
||||
|
||||
class GenerationIntegrationTestsMixin:
|
||||
|
||||
# To be populated by the child classes
|
||||
framework_dependent_parameters = {
|
||||
"AutoModelForSeq2SeqLM": None,
|
||||
|
||||
@@ -17,8 +17,8 @@
|
||||
import unittest
|
||||
|
||||
import numpy as np
|
||||
|
||||
from parameterized import parameterized
|
||||
|
||||
from transformers import is_tf_available
|
||||
from transformers.testing_utils import require_tf
|
||||
|
||||
|
||||
@@ -36,7 +36,6 @@ if is_tf_available():
|
||||
|
||||
@require_tf
|
||||
class UtilsFunctionsTest(unittest.TestCase):
|
||||
|
||||
# tests whether the top_k_top_p_filtering function behaves as expected
|
||||
def test_top_k_top_p_filtering(self):
|
||||
logits = tf.convert_to_tensor(
|
||||
@@ -133,7 +132,6 @@ class UtilsFunctionsTest(unittest.TestCase):
|
||||
|
||||
@require_tf
|
||||
class TFGenerationIntegrationTests(unittest.TestCase, GenerationIntegrationTestsMixin):
|
||||
|
||||
# setting framework_dependent_parameters needs to be gated, just like its contents' imports
|
||||
if is_tf_available():
|
||||
framework_dependent_parameters = {
|
||||
|
||||
@@ -1413,7 +1413,6 @@ class GenerationTesterMixin:
|
||||
def test_contrastive_generate(self):
|
||||
# check `generate()` and `contrastive_search()` are equal
|
||||
for model_class in self.all_generative_model_classes:
|
||||
|
||||
# won't fix: FSMT and Reformer have a different cache variable type (and format).
|
||||
if any(model_name in model_class.__name__.lower() for model_name in ["fsmt", "reformer"]):
|
||||
return
|
||||
@@ -1435,7 +1434,6 @@ class GenerationTesterMixin:
|
||||
|
||||
def test_contrastive_generate_dict_outputs_use_cache(self):
|
||||
for model_class in self.all_generative_model_classes:
|
||||
|
||||
# won't fix: FSMT and Reformer have a different cache variable type (and format).
|
||||
if any(model_name in model_class.__name__.lower() for model_name in ["fsmt", "reformer"]):
|
||||
return
|
||||
@@ -1661,7 +1659,6 @@ class GenerationTesterMixin:
|
||||
|
||||
@require_torch
|
||||
class UtilsFunctionsTest(unittest.TestCase):
|
||||
|
||||
# tests whether the top_k_top_p function behaves as expected
|
||||
def test_top_k_top_p_filtering(self):
|
||||
logits = torch.tensor(
|
||||
@@ -1792,7 +1789,6 @@ class UtilsFunctionsTest(unittest.TestCase):
|
||||
|
||||
@require_torch
|
||||
class GenerationIntegrationTests(unittest.TestCase, GenerationIntegrationTestsMixin):
|
||||
|
||||
# setting framework_dependent_parameters needs to be gated, just like its contents' imports
|
||||
if is_torch_available():
|
||||
framework_dependent_parameters = {
|
||||
|
||||
Reference in New Issue
Block a user