hPfSSKJrJr SSKrSSKrSSKJr SSKJ r "SS\R5r g))OptionalUnionNspaces) GymStepReturnc >^\rSrSrSrSS\S\S\S\S\4 U4SjjjrS\S\S S 4S jrS \ \ \ R44S jr SS jrS\\\ R4S \4SjrSS\ S S 4SjjrS S S.S\\S\\ S \\ \ \ R4\ 44U4SjjjrSrU=r$)SimpleMultiObsEnv u  Base class for GridWorld-based MultiObs Environments 4x4 grid world. .. code-block:: text ____________ | 0 1 2 3| | 4|¯5¯¯6¯| 7| | 8|_9_10_|11| |12 13 14 15| ¯¯¯¯¯¯¯¯¯¯¯¯¯¯ start is 0 states 5, 6, 9, and 10 are blocked goal is 15 actions are = [left, down, right, up] simple linear state env of 15 states but encoded with a vector and an image observation: each column is represented by a random vector and each row is represented by a random image, both sampled once at creation time. :param num_col: Number of columns in the grid :param num_row: Number of rows in the grid :param random_start: If true, agent starts in random position :param channel_last: If true, the image will be channel last, else it will be channel first num_colnum_row random_startdiscrete_actions channel_lastc >[TU]5 SUlU(a /SQUlO /SQUlX0lX@lU(a[ R"S5UlO[ R"SSS5Ul[ R"[ R"SSUR4[RS9[ R"SS UR[RS9S .S 9UlSUlS UlS UlSUl/SQUlUR)5 Xl/UlUR/X5 [1UR,5S- Ulg)N)@r)rrrrr)rdtypevecimgrd)leftdownrightup)super__init__ vector_sizeimg_sizer rrDiscrete action_spaceBoxDictnpfloat64uint8observation_spacecount max_countlogstate action2strinit_possible_transitionsr state_mappinginit_state_mappinglen max_state)selfr r r rr __class__s h/home/james-whalen/.local/lib/python3.13/site-packages/stable_baselines3/common/envs/multi_input_envs.pyr"SimpleMultiObsEnv.__init__&s  'DM'DM( 0  & 2D  & 1a 6D !'zz!Q)9)9(;2::Nzz!S$--rxxH"    9 &&( :< 1T//014returnNcx[RRXR45n[RRSSUSS4[RS9n[ U5HPn[ U5H>nUR RX5XFRUR5S.5 M@ MR g)z Initializes the state_mapping array which holds the observation values for each state :param num_col: Number of columns. :param num_row: Number of rows. rrrrrN) r)randomr#randintr+ranger3appendreshaper$)r7r r col_vecsrow_imgsijs r9r4$SimpleMultiObsEnv.init_state_mappingQs99##W.>.>$?@99$$Qgr2->bhh$OwA7^""))(+hkFYFYZ^ZgZgFh*ij$ r;c4URUR$)zd Uses the state to get the observation mapping. :return: observation dict {'vec': ..., 'img': ...} )r3r0r7s r9get_state_mapping#SimpleMultiObsEnv.get_state_mappingas !!$**--r;cL/SQUl/SQUl/SQUl/SQUlg)aS Initializes the transitions of the environment The environment exploits the cardinal directions of the grid by noting that they correspond to simple addition and subtraction from the cell id within the grid - up => means moving up a row => means subtracting the length of a column - down => means moving down a row => means adding the length of a column - left => means moving left by one => means subtracting 1 - right => means moving right by one => means adding 1 Thus one only needs to specify in which states each action is possible in order to define the transitions of the environment )r )rrrN )rrrM rOrP)rrRrUrSrTrQN) left_possible down_possibleright_possible up_possiblerIs r9r2+SimpleMultiObsEnv.init_possible_transitionsis$3030r;actioncdUR(d[R"U5nU=RS- slURnSnURUR ;aUS:XaU=RS-slOURUR ;a&US:Xa U=RUR- slOuURUR;aUS:XaU=RS- slO?URUR;a%US:XaU=RUR-slURUR:HnU(aSOUnURUR:nUnSURUSUS UR3Ul UR5X6US U04$) aK Run one timestep of the environment's dynamics. When end of episode is reached, you are responsible for calling `reset()` to reset this environment's state. Accepts an action and returns a tuple (observation, reward, terminated, truncated, info). :param action: :return: tuple (observation, reward, terminated, truncated, info). rgrrMrNg?zWent z in state z, got to state got_to_end)rr)argmaxr-r0rVrWr rXrYr6r.r1r/rJ)r7r[ prev_staterewardr] truncated terminateds r9stepSimpleMultiObsEnv.step|sP$$YYv&F a ZZ  ::++ +! JJ!OJ ZZ4-- -&A+ JJ$,, &J ZZ4.. .6Q; JJ!OJ ZZ4++ +! JJ$,, &JZZ4>>1 "JJ/  4??623:j\Y]YcYcXde%%'YWaHbbbr;modec.[UR5 g)z2 Prints the log of the environment. :param mode: N)printr/)r7res r9renderSimpleMultiObsEnv.renders dhhr;)seedoptionsrjrkc>Ub [TU]US9 SUlUR(dSUlO/[ R RSUR5UlURUR04$)z Resets the environment state and step count and returns reset observation. :param seed: :return: observation dict {'vec': ..., 'img': ...} )rjr) r!resetr-r r0r)r>r?r6r3)r7rjrkr8s r9rmSimpleMultiObsEnv.resetsf   GMtM $   DJ**1dnn=DJ!!$**-r11r;)r1r&r-rrWr$rVr/r.r6r r,r rXr0r3rYr#)rrTTT)r<N)human)__name__ __module__ __qualname____firstlineno____doc__intboolr"r4dictstrr)ndarrayrJr2rrrcrhrtuplerm__static_attributes__ __classcell__)r8s@r9r r s:!!%! )5)5)5 )5  )5  )5)5Vk#kkk .4RZZ#8.1&#c5bjj1#cm#cJ3T.2T2Xc]2HTN2V[\`adfhfpfpap\qsw\wVx22r;r ) typingrr gymnasiumgymnumpyr)r%stable_baselines3.common.type_aliasesrEnvr r;r9rs&"?m2m2r;