__init__.py 950 B

12345678910111213141516171819202122232425262728293031323334353637383940
  1. from cvargparse.argument import Argument
  2. from cvargparse.argument import Argument as Arg
  3. from cvargparse.argument import FileArgument
  4. from cvargparse.factory import ArgFactory
  5. from cvargparse.factory import BaseFactory
  6. from cvargparse.parser.base import BaseParser
  7. from cvargparse.parser.gpu_parser import GPUParser
  8. from cvargparse.parser.mode_parser import ModeParserFactory
  9. from cvargparse.utils.dataclass import cvdataclass
  10. from cvargparse.utils.dataclass import Choices
  11. __all__ = [
  12. "Arg",
  13. "ArgFactory",
  14. "Choices",
  15. "Argument",
  16. "BaseFactory",
  17. "BaseParser",
  18. "cvdataclass",
  19. "FileArgument",
  20. "GPUParser",
  21. "ModeParserFactory",
  22. ]
  23. if __name__ == '__main__':
  24. @cvdataclass
  25. class Args:
  26. group_name = "args"
  27. train_samples: float = None
  28. test_samples: int = 2
  29. arg2: str = "something"
  30. arg3: Choices([1, 2, 3], int) = 1
  31. is_arg4: bool = False
  32. parser = BaseParser(Args(test_samples=20))
  33. print(parser.parse_args("--is_arg4".split()))