tests/utils/test_model_output.py · ede051f1b85a33d2e0576b48042a58dc5332ed70 · chenpangpang / transformers

"tests/models/bart/test_tokenization_bart.py" did not exist on "e73a3e1891775a915846cc0f24b7e9a26d6688fb"

Add missing Maskformer dataclass decorator, add dataclass check in ModelOutput... · d7bd325b

Craig Chan authored Sep 14, 2023


Add missing Maskformer dataclass decorator, add dataclass check in ModelOutput for subclasses (#25638)

* Add @dataclass to MaskFormerPixelDecoderOutput

* Add dataclass check if subclass of ModelOutout

* Use unittest assertRaises rather than pytest per contribution doc

* Update src/transformers/utils/generic.py per suggested change
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

---------
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

d7bd325b

test_model_output.py 5.61 KB

Replace test_model_output.py