o h?@s8ddlZddlmZGdddeZGdddeZdS)N)Actionc.eZdZdZd d fdd Zd ddZZS) enva Get argument values from ``PET_{dest}`` before defaulting to the given ``default`` value. For flags (e.g. ``--standalone``) use ``check_env`` instead. .. note:: when multiple option strings are specified, ``dest`` is the longest option string (e.g. for ``"-f", "--foo"`` the env var to set is ``PET_FOO`` not ``PET_F``) Example: :: parser.add_argument("-f", "--foo", action=env, default="bar") ./program -> args.foo="bar" ./program -f baz -> args.foo="baz" ./program --foo baz -> args.foo="baz" PET_FOO="env_bar" ./program -f baz -> args.foo="baz" PET_FOO="env_bar" ./program --foo baz -> args.foo="baz" PET_FOO="env_bar" ./program -> args.foo="env_bar" parser.add_argument("-f", "--foo", action=env, required=True) ./program -> fails ./program -f baz -> args.foo="baz" PET_FOO="env_bar" ./program -> args.foo="env_bar" PET_FOO="env_bar" ./program -f baz -> args.foo="baz" NFreturnc sBd|}tj||}|rd}tjd|||d|dS)NPET_F)destdefaultrequired)upperosenvirongetsuper__init__)selfrrr kwargsenv_name __class__r s/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/distributed/argparse_util.pyr,s z env.__init__cCst||j|dSN)setattrrrparser namespacevalues option_stringr r r__call__9sz env.__call__)NFrNr__name__ __module__ __qualname____doc__rr __classcell__r r rrr s rcr) check_enva Check whether the env var ``PET_{dest}`` exists before defaulting to the given ``default`` value. Equivalent to ``store_true`` argparse built-in action except that the argument can be omitted from the commandline if the env var is present and has a non-zero value. .. note:: it is redundant to pass ``default=True`` for arguments that use this action because a flag should be ``True`` when present and ``False`` otherwise. Example: :: parser.add_argument("--verbose", action=check_env) ./program -> args.verbose=False ./program --verbose -> args.verbose=True PET_VERBOSE=1 ./program -> args.verbose=True PET_VERBOSE=0 ./program -> args.verbose=False PET_VERBOSE=0 ./program --verbose -> args.verbose=True Anti-pattern (don't do this): :: parser.add_argument("--verbose", action=check_env, default=True) ./program -> args.verbose=True ./program --verbose -> args.verbose=True PET_VERBOSE=1 ./program -> args.verbose=True PET_VERBOSE=0 ./program -> args.verbose=False FrNc sLd|}tttj||rdnd}tjd|d|dd|dS)Nr10Tr)rconstrnargsr )r boolintr r rrr)rrrrrrr rrbs zcheck_env.__init__cCst||j|jdSr)rrr)rr r rrgszcheck_env.__call__)Frrr r r rrr&=s$r&)r argparserrr&r r r rs 0