Replace assertion with ValueError exception (#14098)

7af55d3a · David del Río Medina · GitHub · f00bceab · 7af55d3a
Unverified Commit 7af55d3a authored Oct 21, 2021 by David del Río Medina Committed by GitHub Oct 21, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 3 deletions

src/transformers/hf_argparser.py src/transformers/hf_argparser.py +2 -3

No files found.
--- a/src/transformers/hf_argparser.py
+++ b/src/transformers/hf_argparser.py
@@ -133,9 +133,8 @@ class HfArgumentParser(ArgumentParser):
            ):
                kwargs["nargs"] = "+"
                kwargs["type"] = field.type.__args__[0]
-                assert all(
+                if not all(x == kwargs["type"] for x in field.type.__args__):
-                    x == kwargs["type"] for x in field.type.__args__
+                    raise ValueError(f"{field.name} cannot be a List of mixed types")
-                ), f"{field.name} cannot be a List of mixed types"
                if field.default_factory is not dataclasses.MISSING:
                    kwargs["default"] = field.default_factory()
                elif field.default is dataclasses.MISSING: