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:
@@ -320,7 +320,6 @@ class BloomModelTester:
|
||||
|
||||
@require_torch
|
||||
class BloomModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase):
|
||||
|
||||
all_model_classes = (
|
||||
(
|
||||
BloomModel,
|
||||
@@ -450,7 +449,6 @@ class BloomModelTest(ModelTesterMixin, GenerationTesterMixin, unittest.TestCase)
|
||||
@slow
|
||||
@require_torch_gpu
|
||||
def test_batch_generation_padd(self):
|
||||
|
||||
path_560m = "bigscience/bloom-560m"
|
||||
model = BloomForCausalLM.from_pretrained(path_560m, use_cache=True, revision="gs555750").cuda()
|
||||
model = model.eval()
|
||||
|
||||
@@ -25,7 +25,6 @@ from ...test_tokenization_common import TokenizerTesterMixin
|
||||
|
||||
@require_tokenizers
|
||||
class BloomTokenizationTest(TokenizerTesterMixin, unittest.TestCase):
|
||||
|
||||
slow_tokenizer_class = None
|
||||
rust_tokenizer_class = BloomTokenizerFast
|
||||
tokenizer_class = BloomTokenizerFast
|
||||
|
||||
Reference in New Issue
Block a user