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:
@@ -29,6 +29,7 @@ from ...test_modeling_flax_common import FlaxModelTesterMixin, ids_tensor, rando
|
||||
if is_flax_available():
|
||||
import jax
|
||||
import jax.numpy as jnp
|
||||
|
||||
from transformers.modeling_flax_pytorch_utils import (
|
||||
convert_pytorch_state_dict_to_flax,
|
||||
load_flax_weights_in_pytorch_model,
|
||||
@@ -181,7 +182,6 @@ class FlaxGPTNeoModelTester:
|
||||
|
||||
@require_flax
|
||||
class FlaxGPTNeoModelTest(FlaxModelTesterMixin, FlaxGenerationTesterMixin, unittest.TestCase):
|
||||
|
||||
all_model_classes = (FlaxGPTNeoModel, FlaxGPTNeoForCausalLM) if is_flax_available() else ()
|
||||
all_generative_model_classes = (FlaxGPTNeoForCausalLM,) if is_flax_available() else ()
|
||||
|
||||
|
||||
@@ -372,7 +372,6 @@ class GPTNeoModelTester:
|
||||
|
||||
@require_torch
|
||||
class GPTNeoModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
|
||||
|
||||
all_model_classes = (
|
||||
(GPTNeoModel, GPTNeoForCausalLM, GPTNeoForSequenceClassification) if is_torch_available() else ()
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user