From 9ca3aa01564bb81e1362288a8fdf5ac6e0e63126 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 31 Jul 2023 17:32:05 +0200 Subject: [PATCH] Fix `all_model_classes` in `FlaxBloomGenerationTest` (#25211) fix Co-authored-by: ydshieh --- tests/models/bloom/test_modeling_flax_bloom.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/bloom/test_modeling_flax_bloom.py b/tests/models/bloom/test_modeling_flax_bloom.py index 0e49039afe..91716615de 100644 --- a/tests/models/bloom/test_modeling_flax_bloom.py +++ b/tests/models/bloom/test_modeling_flax_bloom.py @@ -198,7 +198,7 @@ class FlaxBloomModelTest(FlaxModelTesterMixin, unittest.TestCase, FlaxGeneration @slow @require_flax class FlaxBloomGenerationTest(unittest.TestCase): - all_model_classes = (FlaxBloomForCausalLM) if is_flax_available() else () + all_model_classes = (FlaxBloomForCausalLM,) if is_flax_available() else () all_generative_model_classes = () if is_flax_available() else () def setUp(self):