Clarify the documentation generation logic for repeat with dtype

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jun 11, 2024 · 8bc99d2 · 8bc99d2
1 parent 5c74333
commit 8bc99d2
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/dargs/dargs.py b/dargs/dargs.py
@@ -679,16 +679,13 @@ def gen_doc_body(self, path: list[str] | None = None, **kwargs) -> str:
                 }
                 allowed_types = []
                 allowed_element = []
-                if list in unsubscripted_dtype:
-                    allowed_types.append("list")
-                    allowed_element.append("element")
-                elif dict in unsubscripted_dtype:
-                    allowed_types.append("dict")
-                    allowed_element.append("key-value pair")
-                else:
-                    raise ValueError(
-                        "When `repeat` is True, `dtype` should contain `dict` OR `list`."
-                    )
+                if list in unsubscripted_dtype or dict in unsubscripted_dtype:
+                    if list in unsubscripted_dtype:
+                        allowed_types.append("list")
+                        allowed_element.append("element")
+                    if dict in unsubscripted_dtype:
+                        allowed_types.append("dict")
+                        allowed_element.append("key-value pair")
                 body_list.append(
                     f"This argument takes a {' or '.join(allowed_types)} with "
                     f"each {' or '.join(allowed_element)} containing the following: \n"