소스 검색

minor fix in the group_name definition

Dimitri Korsch 3 년 전
부모
커밋
f5581ddf17
2개의 변경된 파일6개의 추가작업 그리고 4개의 파일을 삭제
  1. 3 2
      cvargparse/__init__.py
  2. 3 2
      cvargparse/parser/base.py

+ 3 - 2
cvargparse/__init__.py

@@ -29,11 +29,12 @@ if __name__ == '__main__':
 	class Args:
 		group_name = "args"
 
-		arg1: float = None
+		train_samples: float = None
+		test_samples: int = 2
 		arg2: str = "something"
 
 		arg3: Choices([1, 2, 3], int) = 1
 		is_arg4: bool = False
 
-	parser = BaseParser(Args)
+	parser = BaseParser(Args(test_samples=20))
 	print(parser.parse_args("--is_arg4".split()))

+ 3 - 2
cvargparse/parser/base.py

@@ -86,8 +86,9 @@ class BaseParser(LoggerMixin, argparse.ArgumentParser):
 
 		elif is_dataclass(arglist):
 			self._dataclass_instance = arglist
-			arglist, group_name = get_arglist(arglist)
-
+			arglist, _group_name = get_arglist(arglist)
+			if group_name is None:
+				group_name = _group_name
 
 		if group_name is None:
 			group = self