Unverified Commit 05a12a09 authored by Santiago Castro's avatar Santiago Castro Committed by GitHub
Browse files

Fix `HfArgumentParser` when passing a generator (#15758)

* Fix `HfArgumentParser` when passing a generator

* Add missing import

* Always convert `dataclass_types` into a list
parent db57bb2b
...@@ -66,7 +66,7 @@ class HfArgumentParser(ArgumentParser): ...@@ -66,7 +66,7 @@ class HfArgumentParser(ArgumentParser):
super().__init__(**kwargs) super().__init__(**kwargs)
if dataclasses.is_dataclass(dataclass_types): if dataclasses.is_dataclass(dataclass_types):
dataclass_types = [dataclass_types] dataclass_types = [dataclass_types]
self.dataclass_types = dataclass_types self.dataclass_types = list(dataclass_types)
for dtype in self.dataclass_types: for dtype in self.dataclass_types:
self._add_dataclass_arguments(dtype) self._add_dataclass_arguments(dtype)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment