Skip to content

Commit 9ca3aa0

Browse files
authored
Fix all_model_classes in FlaxBloomGenerationTest (#25211)
fix Co-authored-by: ydshieh <[email protected]>
1 parent 59dcea3 commit 9ca3aa0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/bloom/test_modeling_flax_bloom.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def test_model_from_pretrained(self):
198198
@slow
199199
@require_flax
200200
class FlaxBloomGenerationTest(unittest.TestCase):
201-
all_model_classes = (FlaxBloomForCausalLM) if is_flax_available() else ()
201+
all_model_classes = (FlaxBloomForCausalLM,) if is_flax_available() else ()
202202
all_generative_model_classes = () if is_flax_available() else ()
203203

204204
def setUp(self):

0 commit comments

Comments
 (0)