&`iK^ddlZddlmZddlmZddlmZeGddeZdS)N)Optional) EnvConfigDict) DeveloperAPIceZdZdZ ddedededed eed ef d Z ddeedeedeedeed eed eed dfd Z de d dfdZ fdZ xZ S) EnvContextaKWraps env configurations to include extra rllib metadata. These attributes can be used to parameterize environments per process. For example, one might use `worker_index` to control which data file an environment reads in on initialization. RLlib auto-sets these attributes when constructing registered envs. rFN env_config worker_index vector_indexremote num_workersrecreated_workerct||||_||_||_||_||_dS)aInitializes an EnvContext instance. Args: env_config: The env's configuration defined under the "env_config" key in the Algorithm's config. worker_index: When there are multiple workers created, this uniquely identifies the worker the env is created in. 0 for local worker, >0 for remote workers. vector_index: When there are multiple envs per worker, this uniquely identifies the env index within the worker. Starts from 0. remote: Whether individual sub-environments (in a vectorized env) should be @ray.remote actors or not. num_workers: The total number of (remote) workers in the set. 0 if only a local worker exists. recreated_worker: Whether the worker that holds this env is a recreated one. This means that it replaced a previous (failed) worker when `restart_failed_env_runners=True` in the Algorithm's config. N)dict__init__r r r r r selfrr r r r r s m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/rllib/env/env_context.pyrzEnvContext.__init__sG: dJ''')( & 0returnct|tj|n|||n|j||n|j||n|j||n|j||n|jS)a]Returns a copy of this EnvContext with some attributes overridden. Args: env_config: Optional env config to use. None for not overriding the one from the source (self). worker_index: Optional worker index to use. None for not overriding the one from the source (self). vector_index: Optional vector index to use. None for not overriding the one from the source (self). remote: Optional remote setting to use. None for not overriding the one from the source (self). num_workers: Optional num_workers to use. None for not overriding the one from the source (self). recreated_worker: Optional flag, indicating, whether the worker that holds the env is a recreated one. This means that it replaced a previous (failed) worker when `restart_failed_env_runners=True` in the Algorithm's config. Returns: A new EnvContext object as a copy of self plus the provided overrides. )rcopydeepcopyr r r r r rs rcopy_with_overrideszEnvContext.copy_with_overrides9sy>)3)?DM* % % %T(4LL$:K(4LL$:K(FFdk&2KK8H 0 <  $BW    rdefaultscL|D]\}}||vr|||<dS)aSets missing keys of self to the values given in `defaults`. If `defaults` contains keys that already exist in self, don't override the values with these defaults. Args: defaults: The key/value pairs to add to self, but only for those keys in `defaults` that don't exist yet in self. .. testcode:: :skipif: True from ray.rllib.env.env_context import EnvContext env_ctx = EnvContext({"a": 1, "b": 2}, worker_index=0) env_ctx.set_defaults({"a": -42, "c": 3}) print(env_ctx) .. testoutput:: {"a": 1, "b": 2, "c": 3} N)items)rrkeyvalues r set_defaultszEnvContext.set_defaultsas>,#..** " "JC$!S  " "rc tddd|jd|jd|jd|jzdzS)Nz , worker=/z , vector_idx=z , remote=})super__str__r r r r )r __class__s rr%zEnvContext.__str__{s| GGOO  crc "D$+DDd.>DD+DD6:kDD DFI J r)rFNF)NNNNNN)__name__ __module__ __qualname____doc__rintboolrrrrrr% __classcell__)r&s@rrrsU%)!&$1$1!$1$1 $1  $1 c] $1$1$1$1$1P/3&*&*!%%)+/& & ]+& sm& sm &  & c] & #4.&  & & & & P"T"d""""4         rr) rtypingrray.rllib.utils.typingrray.util.annotationsrrrrrr2s 000000------w w w w w w w w w w r