a ==ic1@s8ddlZddlmZGdddeZGdddeZdS)N)Actioncs2eZdZdZd ddfdd Zd ddZZS) enva Gets 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||}|r$d}tjf|||d|dS)NPET_F)destdefaultrequired)upperosenvirongetsuper__init__)selfrrr kwargsenv_name __class__p/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/torch/distributed/argparse_util.pyr+s z env.__init__cCst||j|dSN)setattrrrparser namespacevalues option_stringrrr__call__8sz env.__call__)NF)N__name__ __module__ __qualname____doc__rr __classcell__rrrrr s rcs2eZdZdZd ddfdd Zd ddZZS) check_enva For flags, checks 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 FNrc sLd|}tttj||r"dnd}tjf|d|dd|dS)Nr10Tr)rconstrnargs)r boolintr r r rr)rrrrrrrrraszcheck_env.__init__cCst||j|jdSr)rrr(rrrrrfszcheck_env.__call__)F)Nrrrrrr%<s$r%)r argparserrr%rrrrs 0