hSSKrSSKJrJrJrJr SSKrSSKJ r SSK J r SSK J r JrJr SSKJr S\R$S\\R(S \\R(4S jrS\R$S\\R(S \4S jrSS \\\S \R$44S\S\\S\S\\S\\\R$/\R$4S\\\\4S\\\\ \4S\\\\4S\\\\4S\\\\4S \4SjjrSS \\\S \R$44S\S\\S\S\\S\\\\4S\\\\4S\\\\ \\4S\\\\4S\\\\4S \4Sjjrg)N)AnyCallableOptionalUnion) AtariWrapper)Monitor) DummyVecEnv SubprocVecEnvVecEnv) _patch_envenv wrapper_classreturncUn[U[R5(a?[X!5(aU$URn[U[R5(aM?g)z Retrieve a ``VecEnvWrapper`` object by recursively searching. :param env: Environment to unwrap :param wrapper_class: Wrapper to look for :return: Environment unwrapped till ``wrapper_class`` if it has been wrapped with it N) isinstancegymWrapperr )r renv_tmps [/home/james-whalen/.local/lib/python3.13/site-packages/stable_baselines3/common/env_util.pyunwrap_wrapperr sKG Wckk * * g - -N++ Wckk * * c[X5SL$)z Check if a given environment has been wrapped with a given wrapper. :param env: Environment to check :param wrapper_class: Wrapper class to look for :return: True if environment has been wrapped with ``wrapper_class``. N)r)r rs r is_wrappedrs # -T 99renv_id.n_envsseed start_index monitor_dir env_kwargs vec_env_clsvec_env_kwargsmonitor_kwargswrapper_kwargsc b^^^^^^ ^ T=(d 0mU=(d 0nT =(d 0m T =(d 0m UceS[S[/[R44UUUU UUU 4Sjjn Uc[nU"[ U5V s/sH o"X-5PM sn 40UD6n U R T5 U $s sn f)a Create a wrapped, monitored ``VecEnv``. By default it uses a ``DummyVecEnv`` which is usually faster than a ``SubprocVecEnv``. :param env_id: either the env ID, the env class or a callable returning an env :param n_envs: the number of environments you wish to have in parallel :param seed: the initial seed for the random number generator :param start_index: start rank index :param monitor_dir: Path to a folder where the monitor files will be saved. If None, no file will be written, however, the env will still be wrapped in a Monitor wrapper to provide additional information about training. :param wrapper_class: Additional wrapper to use on the environment. This can also be a function with single argument that wraps the environment in many things. Note: the wrapper specified by this parameter will be applied after the ``Monitor`` wrapper. if some cases (e.g. with TimeLimit wrapper) this can lead to undesired behavior. See here for more details: https://github.com/DLR-RM/stable-baselines3/issues/894 :param env_kwargs: Optional keyword argument to pass to the env constructor :param vec_env_cls: A custom ``VecEnv`` class constructor. Default: None. :param vec_env_kwargs: Keyword arguments to pass to the ``VecEnv`` class constructor. :param monitor_kwargs: Keyword arguments to pass to the ``Monitor`` class constructor. :param wrapper_kwargs: Keyword arguments to pass to the ``Wrapper`` class constructor. :return: The wrapped environment rankrc H>^S[R4UUUUUUUU4SjjnU$)Nrc(>TceT ceTce[T[5(a.SS0nURT5 [R"T40UD6nOT"S0TD6n[ U5nTbURRTT-5 Tb)[RRT[T55OSnUbTb[R"TSS9 [U4SU0TD6nT b T "U40T D6nU$![ a [R"T40TD6nNf=f)N render_mode rgb_arrayT)exist_okfilename)rstrupdatermake TypeErrorr action_spacerospathjoinmakedirsr) kwargsr monitor_pathrrrr"r%rrr#s r_init-make_vec_env..make_env.._initSs)!- --!- --) ))&#&&'5 j)9((64V4C*z* o  %%dTk2DOCZ277<< SY?`dL'K,C K$7#G GGC(#C:>:J+!9((68Z8C9sC--!DD)rEnv) r%r8rrrr"rrr#s ` rmake_envmake_vec_env..make_envRs ! sww! ! F r)intrrr:r ranger)rrrrrrrr r!r"r#r;ivec_envs` ` ``` `` r make_vec_envrA'sJ!rJ#)rN#)rN#)rN  %% %$s$xCGG 4$$N! eFmLm8AO4mL_P^_G LL NMs>B,c 0[UUUUU[UUUU US9 $)a Create a wrapped, monitored VecEnv for Atari. It is a wrapper around ``make_vec_env`` that includes common preprocessing for Atari games. :param env_id: either the env ID, the env class or a callable returning an env :param n_envs: the number of environments you wish to have in parallel :param seed: the initial seed for the random number generator :param start_index: start rank index :param monitor_dir: Path to a folder where the monitor files will be saved. If None, no file will be written, however, the env will still be wrapped in a Monitor wrapper to provide additional information about training. :param wrapper_kwargs: Optional keyword argument to pass to the ``AtariWrapper`` :param env_kwargs: Optional keyword argument to pass to the env constructor :param vec_env_cls: A custom ``VecEnv`` class constructor. Default: None. :param vec_env_kwargs: Keyword arguments to pass to the ``VecEnv`` class constructor. :param monitor_kwargs: Keyword arguments to pass to the ``Monitor`` class constructor. :return: The wrapped environment ) rrrrrrr r!r"r#)rAr) rrrrrr#rr r!r"s rmake_atari_envrCs2<  "%%%  r) NrNNNNNNN) rDNrNNNNNN)r2typingrrrr gymnasiumr'stable_baselines3.common.atari_wrappersr stable_baselines3.common.monitorr stable_baselines3.common.vec_envr r r *stable_baselines3.common.vec_env.patch_gymr r:typerrboolrr-r=dictrArCr,rrrNs 11@4OOA  S[[0A hs{{F[  :CGG:D,=:$:!%<@+/EI/3/3/3Y #xSWW -- .Y Y 3-Y Y # Y HcggY%789 Yc3h(Y$u[-%?@ABYT#s(^,YT#s(^,YT#s(^,Y Y|!%/3+/KO/3/3* #xSWW -- .* * 3-* * # * T#s(^, *c3h(*%[ 14 3F FGH*T#s(^,*T#s(^,* *r