From 8c842e078fe65559fcdec541f07349cb7716df33 Mon Sep 17 00:00:00 2001
From: Tomasz Sternal <sternal.tomasz@gmail.com>
Date: Thu, 18 Jan 2024 23:41:08 +0100
Subject: [PATCH] Add drowing lines, change color to green

---
 mo_gymnasium/envs/fruit_tree/assets/agent.png | Bin 0 -> 9329 bytes
 mo_gymnasium/envs/fruit_tree/assets/node.png  | Bin 0 -> 4001 bytes
 mo_gymnasium/envs/fruit_tree/fruit_tree.py    | 126 +++++++++++++++++-
 3 files changed, 122 insertions(+), 4 deletions(-)
 create mode 100644 mo_gymnasium/envs/fruit_tree/assets/agent.png
 create mode 100644 mo_gymnasium/envs/fruit_tree/assets/node.png

diff --git a/mo_gymnasium/envs/fruit_tree/assets/agent.png b/mo_gymnasium/envs/fruit_tree/assets/agent.png
new file mode 100644
index 0000000000000000000000000000000000000000..8027fcb1aa8ed57aac49d23bd89a4b129764d217
GIT binary patch
literal 9329
zcmZ{K1zeNe`}Z~`Zh#6yNx@0iV3dT!=njF0M!LHMBu7c7z@$M^MQJ1yM7mo*q!H;B
zMA82}p63_u`+wi@*=P5C&bjJ)#kuZt6RoMPaE*wD2m}IMQ&N=G0)fCR_%DnQ7#Z#m
zp9OAA?PX*%m1Ja)njWq;_D<Fy5L>c$lA4P21JcMlDsbiVkoWm8Jt=dtwZaO6s3mFD
zs=1CG2Z}Pf*YdTF<U59P$HWZAb+9?v2}xLJ46-bzAL*9rk(HIED}WyO*ni*iD~W)J
zVDZ=^Wa=QT`JXsPclKQecJt@4^zzgr$JvLG(o$9Xe+A3`(49jFSx&u@Hr<{`_O(yk
zqu@(St9jS@`_5>l;#aToTS|$q<HV*}!exy7tPc|08t$sV@91a0IEe1#6qNY0FWWA$
z<hZ0BsHOswgg5Ka9P1F+j%A&%U!VFuWOc6zS!e2zgbXs4EtZF^Fw2J+eEq#~5O91p
zjj4+NAj&?~(+_>a79*gv1TygIawpc{ySq{4a`7|LTkWL%I9cq(_*42`fiRTrr?>N4
zriEuuUJ@5m&z@R4Tl>3sJ5+Z$%emdo{OU+zD4+k>Hq%vZ{1v^-5rJ{~0$63vT4uq-
zeF8w-!dhR+MokUG0bIjC5HJEn09=8AKhP6<5cKjI1Ohw+d<aq^L05ra1n||&hx}_6
z%#u&=pKJU=DQy`gCE!=v(!<)?#naB!tJHDP4+MgU+w16i>8q)VTDm%On_IbBSabV2
zyWvSdV!oolrL(n{InvkJ$;DIDSDfK;g(z^1ALd~|UM}%+6lc&^(?rU+dRQaxaih7>
z3=%|0BvQ=7%0^U6R{kGy;7**u&dbY9l!wR1$A{a8pWD^LmWNkFM1%*;$HT{m0#=|r
z{an1veNir+jDIuv7av(`PfHJbH!piv7bKpqxrM8@mpB6hUeLckf9q-OYyTffE}s9G
z1sKSKpW)%<M)Ul?$h_=r{y${+nZL;{?fP4s7@nD^roFGVlYy+gGk_|fngl;uNbFL~
z|C;%aqJLBV<!S99<LV3$dP)4pE&mYz_vHUgcqvl<KO*@={tuD=HS=GTcoRf5JnR8!
z=6DPxc*S`B*VsSn#dz=l{}<qYd-HM>aHj;37|*{$LxRZi;Q1&BgrrcCmD2GA@8tN$
zS^bsMJ-nwoh#7v~Y*W(wIFgQPLJs;U&JN}d?u|*bE0ij9(arD{4E_C*n-*=LSp!A{
z{Cc_Av>DZ<Fk;6LJ&dfBkG6P3L27GqmD={(X700HKQvm~JNT#g<X_^GlP7N*HYJAM
zo%*M4{uE7I_>t}TK5(dUJ(n|*79P@eWYmx=S+M39Xv%XS`&)T(fM(R*(yA|N#%aQ-
zElB)3efH-(uU~G{Vh#Rv!Id+CKi5}YC)SKlOD0~p&l&P@5Hr{e>0QLG_iC3G{T_*A
zRLFHc|C%RFUKtT<BnA66wdtrPGjG$eaE)%}$|NpAQB|62iU50yma@J+gidBDJ1j)p
za^kb2@;QNs*3V*M!m-FRxF&AdHrtm(l)vW7#2R7L&_X%x%XHpV#(T@_F=ugo9Z$Mb
zNyowpZFs&*3RwqYzOLud3S!PZQ$Z2Eg8^{sG~JRnCu$pk7RL@cQ%84Y*CUCLTM3;l
zAp(*w<>?PpS2(NR?rTn^>UjNRSEc3x8<szO!y#0Vb>sjWB}vijGviwpXapBzD)^i<
zFg(uVntD%ZYejPZG2~{+w`cP77B!BK<Xt}CMn03BMVzHECNeF6r;=MP*tQphwvOo^
zUCa$T5v4x&E7Gp{RQ}Bwy+`zmg-($Fr43!!QY%GTv$g$hsezho?oJ6><Y{x0^;a6N
zQI{u5dP_n}N%nsC)@~wB*_s0KQl1~(SQjZUP^OJwUz;22o#7niJBN0~yvDMB@1<@y
zhGln5Pkbz#-|^UP>>lF}<JHD#u@N+Ru7?LNci!*pk-U3e!H#QVdqOT;DBrs2EV{np
zu^8g~m%B<5D1u#T+P36{D_3kIZP@f1=~{z4?Wf2_o#%%Sl!Em1Jhz&PXA`ySiZIFA
zIEx{ts1eatzMlpIyiH^e)H-j|g(YURA-*c_98j!F#Kd^A%$LQ`iAJ!O{Q9I;^OyGb
z)~L{56tW+>fAYGkvrmiu9u>3OoWHu(VT@v?^`JvVurF^<_6@B04xdP<2DRLls%fU+
zHSG@l$jNsoJg@OyiDLbZ&p^FumW+7}+t2N@2TB9)U@nwFt2AVv;-F#l&3+{XUu_a(
z<WenPVJtmrJrgSfx`PX?-a8HCe-pn7)hzfOwPrCtEb?H*nkB+sHb}Jbf(Z4)AHBD>
z1`EBdu!_@mr7{!<PKxwlT$hPY|M*JtZsSHhU54HB1C(zwiR2-1?;EorXUaV|d)hqe
z$qn(pDr*zdjCEgyoNX5=iu;{6+ueN!f7pXU3W+)*Dp6&3db>{c7Oe)&m_s$<A#Ux&
zbv~QX<N8EXE!0Eo%3$1RF9!r?Vm9y8pM+9*xw~J)3f8U8)lcWS_eq@ZoY`(P9X#Y4
z3zbHUJolyBATVZqA$gBT`^LSdA1(H%$>2yMecCKqc!#{%8-czz87o&QEn@FJ2iI_Z
zN^_4_KDiz>cGtLs&AV#Jeii!YbG1#s@#<Y$F-$$qkh(0DpR$})^25dz>S687JT*u#
zh3Nva`|GLo_%FK(n^?FQt<j59>8C-7GBH}UYq14aSlzIwuzD`S(-hN;2p8UsuN$#-
zs~3<Nqd`{+erswb&D6+0q9X72y$GEXJ67B4ar+%R3rJ;+5C%E+l)?@S1CwT*^?3Ry
z?BjrrW`rZ}#__RDg?}PLhVa{*@<Ky>%Twi<R%Hr`eS?!_rw9k$NzO^y+NGHjPd4RF
z_7ww-4HqLD=b6FB((TixI0|z5-d!n+1!Ca{B~M%^reTO&v&GjZ*uKg}PkQho{ipJa
zni|>@xw=?ilczuDtRS%%v%GA^jQ+W~UNP2tpG+FRmL@-(5V1nau@4u&)wgek{jpnj
zni=w}_T*F*{p+2a_yvy@ed}cUI>LCM*16XDbNWFGi)4n0NbrkunK-7IM35UJHh^67
zD8E6+-*F!CjMKb*aZ&V()|Xgj7VaS<d0ug*k&F;r>6)Fh5QnFW1bZ5-d3B!3ShYm-
zz~?C;0>3cmp~_<A#M)R(fI0`IrK+gl1x?gg&{L5ZtsqW%?R&Vkdq!8COsKZGXnpv3
zfaxRdTYGwK;>p)mYg3>WjA-+%&<gUZ&LqvYXA~v!*54U94%I3-m^)%elR~zj%Ju5g
zO-oG$VcY$$&@+^`tC9|Flitq7rC#e}n`gw5LOZ77xmuywy~^=|j=N-ncVy(P$9|Ju
ziBnfHyA63^A)zg5*gqFgI`Eg1>eVx`hcxFS!i{pROqxs(JBdK>I9wvTSin{phaNu(
z`gPS;BONrZi*Un&9#ffS6lVrTi<2AKghv-XjXsoRq+>~9cJJ$0^r`vTm7elZF1N%Q
z8?xg(-SqT|Q4S@`{gFZ5Vy>Iav$!jedhQv~wWbPnVr@_5w!?9uPa<Oyh#nH~Pub5W
z<<Ox7-Wj+>cIne+)#@@e3ZQt_ANuz0yPn*cyG3YJtIb&U=uS}}w+)js1C|U{R`&ZH
z*I=+ew6*H7`|D*P(;kzc)A%>fIx<g4vZ0w!n!?{{)wTPy8NR~Isujg5-w8-fd%Vk^
zj-<Dc3r8?(b9>qM-n1ob&5CV7y(yJ_(C05`c5Vy+|B0{0d*s%~IWm^!L#9vK9>vBZ
z3Cm>DQ^psQ$zJ#GstyV^HQs*BC>LsTGaa;PR>Lg{LVBA!1Z`mNTzAW78-XG~;0Za(
z_wQ3S7k!SJUHkL&l}(TaY)T<eUD(-j|3Y6?qiKb)(iu|F+#&F;&-INpp9RE_m%IW5
z43u}uVB7KYWmgAO+LUcGP)C$uIY}5I=)L+)JE8})sWE6lj!AJvl;r0e<#c{61Wa|Y
zP}k(z_FaxA(LQUNX`9zHKkgyH(cN^n1R`Z1R~Yi&^X_r+OwlXy4MxX=scX1SMe2Rb
z@EBI(OOA_$Dw>f%EFLDMn5IQUtXPI4`Wy>QM1aiIdMdqkgDw?w4joCdoB9GpDXHQN
zw_3N=(av!(P(`FDq&e02BYX1eY?s{k`YQ9}W*xT(3OU(GqM<1)BT13L&u*Lgita{{
z+w)324Z?<NK3yJCi`0M|AlT|QJzA=e$I}t;E#ie(-xpP-;bci5IFnTV?2GZLu3L_J
z>a4chS{gx#H#@jPs<bQ?nJ|zX=zd?!d+CJ1y{#6iHTPGc;0NdMxOl+?Hghx=0*S>y
zTXer%U-iAIP!&yR6E_bI<EH9lJHHD?xaGE8sSu@)EvWKZ^E&byH+*p?v~x}Wl}$dt
zmYcLp<#~%t!X4wNS)9J1Drs@v2hA9*iN?--#WoaVP1huG?nO@oJhk%mleBC_YCm0G
z48koXB*)i&E1<||)kLI~&l9@0?h-wrOehawT111zH&H{b@nda@MnG2kv&ymeZJX;e
zC0&73TQ}}1*g9IFSiUrRN^4|447DE$fUt<F0+=b9!La<*_7_iY3%fZ)HDekWfg0L7
zz91KOvluhI5e4;7Ff&mhU{KO$0?6xE5Y7VIDox4ZluCV6FnAJX4637C+`qbCP}&tQ
zoI*h^yJ3V5FBEWb8IgorWrft%nr-XEp;YmEqQ3jgbUGRYU?xWT_Ej<5ve-Jn{P5SC
z50Luwu0F!YEZ8|xQq+&{mW-Cpmhe=UXN`naM~<wP)?o2e%n+>1&rgL`UVYlPz39-o
zV_<5m=WHVx>Y;@3G606>dfL5gUYyH!oR1}2mMGwbko(%JFv=^h^i<HbYF&_9w73=N
zZvYnO!=T^&W9u^FQbv*uRSlmt-cH`3VV1_bjEw|qf~tV!ERKfv`kvk`Nj5Sns3^J$
zaN|OdnoPxv)IXu1{32j9Sv@PaD37Qcp%^0oKuCw5X=TuPSiGiSAzSJ!-VP=s3qPgo
z;%H(9BYKFz!(5K?Wyj$TyzCFhItw+283|vjQxow6L0$l#u&$~HTX?dtMQEb#;JY5E
z>H{L>j9tYLC>Pd4<vwH_Hfhm*6Dt=kc1zexlnD&zVhvHjiE6@$7~Nva1r1^fZ$9Pv
zk_>`-A#g<eOMI2+w9dIN)WnHE+AAVSuxJ`Y8RsX%d#BJi7?+hfNr@4_LYj2H>ep@w
zvVV_|Jd>Pf{)*FwIf_soawP~88Uh}cJYbN}=L;a7uxQUeHHY5XST~5wlBWdXR)on7
z8n*a3ve?m%Q<2s7jV$tJu{o1p9v}*atw5<!e|z*6opzi`AhW23K!Oo4DGY&KpQbsV
z!|Us0tT`zlcv71g1T0{kkFHDao)Z?Pm<K>YVl|=nt{Xk}sYI90(Q}B3vf#%+(8}Xo
zj5GC_&UKBIZQ^Ef@Nxn9B~zKjo4G^I-_{FUW0gv6-h{EnXi0jsr~}Gh$3~}j9g{q7
zMMuw1jbG)#3&cR7ULjq_w^%wHTPyqHp#Zka3?q7$kBL%;2ArNSr@9+_bi<<sg0dJH
z))2~TnDg1LrVDcWVexAVSw<8b9}|(Mj|BQp9Po?bHTcFdqJbY1Go7Xb4PPQn2%-E=
z<Qb1%vLyKQ@$RRSUAfp#m(-dr_Cqh+J`ts6X6ZF9!r?&4=}-jKw7G6x4~j>Bz+U_k
z62kLTj0$<*bic5!oyn%;Guhuv??J=`Y}3&8*5`(@4`T*FgsKq036AimhO;#uz{HP|
z<M(Oz#Q<s~HXP>VILu*=cy4@?yCBX?09a@XzZ~MT7?Rw2ssIcyC6QvecMQl%zTW3_
z7*gvYk(a?Uw1s2*MHjM+$XHrk=KZE_B?5{oA`KupG_?$>xiNtZ-Iu6o-5n{H;%qOn
zcvNa}jS=rRFzL66{9ESjwwXhWbeBfNW1bM^++wjjdsZ^x5l)s00v`He2n8dQ8gxx#
zV{;hflrPl~0KX}gL-9KFd`_tkIRK(Jsf`ALLK(U+M-OB71jBs+=#lQgh=ccnmEJNT
zS{lz?U&hQJHoY4c_sqmIkMdGUbI9yi;TlEip|%?Q$kq#B{ti}KTPE0LHFfIo(eY9k
z8`xlqVedzU(M}AcqN+qz&CCivKBAUit~4K>{v*Lyh$a`%v`D_qA;mVJQ9zKGiWl!E
zFY*u;Kl!J7J>FqUB0wNLhUP2j+qXGs3`4puJ@FE*uXY%>SwETK2NYopgdGvWvSgpF
zqa}cS05D9#P=cSeQ>D&Kc&C+KszMLu%^FSPryZ*a;r(ZUU5YcW`;-qQWiAk`MT(%z
zaEes7DW4d>wlW~FnqZhzq_VR4=5o6>;A8=K*)0~YuW#{)h1`c`ei0^0y-zKz0$?$s
zZxI{a2{^7z5OVwM5lX@C27v0JsOolK6%%00TiFm4ONK{SJ#fuMUTLYS{|9LEfg53&
z5KmloFA<G2gsD<G1XXhDCO)Fv!Z9_S9j~()8>RRG_L)#NRW2A{#1NSsQR-f{fj|^c
zOM9RJ&D-#Z@*cSmdsgr#K-q0+_fnIZ*zW-2n>dg_Z#^!T2MuA#xzuZf>9d9CZl+1?
z6M(i2Koi$pq;f^Rtq8zESOJr<J=R1#G$R)V$cs?{Wv)i!8kZ4feE<%|jAnuWuj@@V
zR6F=e%?&m^;mJKX?p<d1XMWM|@TrgT=8eXGuXw36ILUdTTn8q3R19{x#GeJcVDu_2
zy)&w&b$kHUr3OR+%g9)vwK>A`RbY<)KOw9Vo7o=z`*tZwQ95}X4p3Dsze29UojI^1
zFzoL&iEY&sbzQiYcN^60IDG!+%V9F=><n<g$wDZg9}+tO2f|Rqq>bhyJ3+ZE{2CHl
zNN%TotCG~vQdNzfHC8=hgB3{nG$1>GeBlkS<(?EXUuDvCgV5o#t}v2d@d^*6ssy}9
zDPKEGHyp0A{aZ}!Ih7C)RUhD9b|VJn99sley4zRrk*kz16Q-NnXW;(}HL0yn@M})M
z*>VoB_zDy9>y<q_=fa%B8WfAa++fWP?c8PHSwNd=feKQdtbpJTC&R}RCKl81{S|^`
zSm3?1)+Jg3SeG3GAeG`jknD>-DlWQa832q&Fpsn?yR8jPym8<abS-<f=hbG32QDfN
zb7RneQ&m2Q_9Yq63Ik5mCDVT?;ux+ZyJxiagcZ+|lo9(dv&JT_w=2=TgnznX^QhB~
zG7PBqY}{DUu8b*f)?l^R@-?&I1{r)xCSgdR-%p8ZO3?+>Cl?PB{4NVoArrp*K3em#
zv1!gtmU%$aRtX4G%+CsWHBQGT`DZS3>kH-VI+d+q`+?QZ>`RwL;o5lPc+^gaYHamh
zukBYor1!n##lWz?6|@BG4i6e%=CM8me1gQf!!ee-yOi?7ye0LPUhBEt2Yp)5ovlmY
z+Hg?y0@{WGQ>s8CRn(2d??*jbc*D?ySnMpHCJxPbq-`my`UXEXlG=uLrk+l#xm$SO
z2htobF0(F&M+&B~$kgPXMd(L6ZQ<}(=V(74<eP}cOPUOp&n^=IXI#&N-D0Q11JDM7
z0H-blE<&!?cKQtTz75fJxj2qr{R8AzQrkCZLOyMSGw!qi#X=iG^@X1cIe%$^IdtNi
z-jl|vQ-V62*ZL4#+<o?X=v72Dl5G4M0O}I#3~2gTMBEl9DxY3O(9n|J-1hkhSqa2z
z_f4VAL6P)9^=|@rs<*zL=gTjijCfQYz`{q9FeZ)lWK>Yo4)@^(>}Qv^m(u8=>F94J
z8}9Gci`!5vTeNqISCGCMaV?4fEKwN!9iWHNvYgsDjD2T(BS4dTpB_+w72C8JVG<R8
zk&ekn>BROD#+4J?AS8O0fcGQ_`pPDpjF}ie@puF{=0UwQc_j3-%%F7+m?lza%V^q8
zBKNFAv)@2*_mlOkm@^dM`<g;TQuoK06y{roxL2&aztyJAd!jFE+%a=Ls^Z&L{zCkS
zN#i37PIRC3w&}FQdM+OICfIiOO>Bmj39*-`R%~zNoTjH|KGipTeM^FNjYcIKsiq5*
zT>r&$?#aW+<js-^Oz?nV{daif7T^37rSh;`rO@U#9zC@jt}3~k<d+2{k_R2JXU(EL
zs=FO{XY?;nlM6DL-1IY$$3O?yLkZ^9@K<nrL&f@Fr4C10BhW%yW%<%R+c@78(Q^ar
zY~QZWPSf0JW!X*M^ltiMP(OWyA3%&A_UlG&#}2*FUCRdCE3g*FJKFp8nO%d|&RMZe
z<Rj@6aXB$|;Tu3Nj@~*{ur&Gz0&J^R0ns3Ji5aFD$XFboI2r%Bz71&=zm44?2?Z)L
z9mLj<hBR7k6rO4lq_;Mntlx<Ni1_5%rY`dPzY+p%f7n6}*^jhmjCPXF8!cTxq63}v
z$n*K-fTBE~*jJqsUEJ~#BNAT$Ad!Ed%VkH&4LGNJX6J3emO|edKmM|1*H(l)W38kd
zCwNHh&B5TxYd}!^AV~btNm;zLFOVeaWMvJTk5}b88CAP)aGPn3w<@@x2mqMeB10-x
zLYwHHNC6K}vyp`=rz8nHc6pu}Ko>&`36`gH(K9ZOF7BMTryCGRkZ|!`<P>1nh+J&g
zeB=}zt%n1i**?x@{vpdcApvSQ&`s2<6K_7sChML}bEHk+{^BW|>-;x61BmChcIU*;
z=lVvf&MU=Brm-vXmt_O1im<EQ^w{oV1+=5gl^@?uttySgKQ(s9w@sN#nIxb>6GF&0
z`s!9zI{NRyca@NyP>W}ZGl|j+sFFvkTHoKaAF}b(4mSdM(gQr6&)9rX8=0_*us2_N
z`{wIA#XDbZ+=1i*rs}+PHGNWR1s|AK!#RP+<`9lAc~%V6lt7l-L7hzMydMcHJpZZe
z6<0p48YgrCUZ?baxB_%3A==RI<_>NB<CSa&6x#QbJrax)R#JC7cY$3go58L6F9mI0
z4sr9$to!Wmji2ZorKRaq=O(9MOd-B&vaiV}M&EN=yid+V&v^9$#{<!sMwaS~VSu(U
zskSL4BOw{z=s9}_OVu3qgVGn09Ri^ZUYBBllYt4bZ>Y)nOzDA_g<jIQS8~55&Rzs8
z!Qo6A$WS1@t!w$)!@E<qYPH0}!<}O}+)02<ABo0(tk(YwmU3GrcPdZd^=7ES6me%P
z&z28ui2=JnR8AYa(Wx;RbC#iPcrOhj>sN=u--5!c%1{cJ5F#oM^KnYV=yW-e!<R%I
zo+@aRP7xHAxPDNVw&<sr!Gz`j-<k}r^mds5Eg!CDqi4eV#^!`3d%E3vd^VT+{34CK
zAP!qN;PmFR9H}}{>Xp9?vwz+`Z}C}7T3)s2El@LH37K>sq^nP8%!Ey^f1jGxdUuul
z{qIg^PJ24I)f14;#(44bIwRxyX+QJoeJ_=hBRij<9_ysqcxW{S(-9z*FeS0wRPl_I
zh4?|~i&L46g%wfFSp~>z1ZH?;t6Hb#Mtb%FAERnIQ)lmwLs<ek<|O6^)vz48U%P3o
zx;xI=wV%Wv0`c%oEq^AkUs0tzCsY+MbM|}U&LZ9QAGNP@V&_sZu7DBB6kSi728RI~
zq0_8H-@-gNq<mZsv!O#k2WqYyvZAptl-YdFrfsSEQmxycGi@d@YIJz<amn~-4tH)U
zu!&`+vYO<U3UCEJdoyx*3iyK%VMvp%KhKJVkhV3ueitMcvh}*UmIqi0pDf?_ROWkQ
z#f`zPeIo(tp}44X6k&XIA&|IVl(OB5ydulyrQsB-d(TOBdpe|Pm;n26uvq1`5wn?2
zWg{62F(`3k$~^t-;ca6v4$^Z*>;PF?U#jn6J#E@dwA_5*Vn%PN#|6TLzw5rC|CR8K
z4?!-zIZzL|w%4Qae0i)7s(++%nmbSmH{(QYntpQ*{je{)cXRhhzOBJ}F5<@34)8ra
zlSUVesEB{@<8y7}h3UY=b*3b9$e=4<$whRr@Q*b^9G7rT&X+Qt%@yI*^_~?jC6fE>
zke}O>9^=2LtP$|Rcj&si_wWZ?uL5hD(Dp8+IX{g_+lrJtfz#hdmfTwWOk@!+icXYI
z2sH*SsBTO8z$oWgz&@}jzYlsx^M@8YV4t{{Pbt#yRJro;@vORC@tBWa4hpTmF_FTt
z?|?dx0-SRtho>iFJbvi7kosqRn77zT(_H<$l#mxq1C2c$V;ZTnt~JmlW>Kbiykw}F
z!>lHxVfu=BkMgbVPRo!bLh%_!jpp~(>RJ0Z`{!iCW8k#gJEfjF@sDDS{+(qB<Mr%P
zbzw2}un^b&LR6|81vByc5md=gO6fm`@^!?bZ;o(K1jD}V!O1lbryx`50}(p40wdKN
z5ldp|Z|$9S#u<&p$M25l6P1wi5O1TePBYKha*78i#tkSq75?>lfH=UTK|10pFX=RA
z{NdKx*@p<K_93sNn<e0x(zMMJhhvp*2QJi$aSe({Ca^e%yaY0~ZZG0kpKASP3CN&A
zyT<25*#*sh6KH579C}VB%*c_KUfh^?>e;r{4?JUlRYG^?L<(3nf+=gd7N9(<6iKu>
zf^atFfkdm7bDth8L1~YbFj%*8DNeU>#Q8q=sQil2TVLv+)rOes8ci@W<IiGui^rPR
zfPJuO_rMDT)_nzbBm6SfyQ5dO9(>JZ`^A-2VVLg0u7T7n>-*D0{0ps^j9&-Z14M!>
zGGF*Suf69uvxXjx@rD(i29Tafl3mz}h`LH9d~lWp-pP@D@b<sY|5u<T@Sf=HdtynC
z9m1!3o}Z;31Fwm`daKmkS-)T2Ta<8%&0+{RTQZJ+qGx>U0Ud<DB0KWt5^m{YN{l^l
zP3iP&rn=h@1=GYaKxXLTF0!Ap|1|Ab!8I&CZ2T!P%&(T8z=M&*zsO={%~SC}39+g2
zuES=}wtn?(+dfFz=?l~5Rp?$^VXh1h7VvNBY)mX%&W;v(Vev%WTp9gmM_9*T;0x(X
zy}O2wju2Lo@;VY6y?gu8pDS)l(u7$kzPVYF)K=)`!s>`W{fYSVzW!F7U<{gCeidja
z)0(@TMAz@B$HZmtjDXlxWym_W@_$5^rvxGR``*Z_{u<C}dHW<H2r;aECpmI1d0m6?
zTG(QHAW<rD<4ARZAS<xRK8#ohmnm7Bx%o4u=Q+;*n0-jxgEfm?b$7LsLuG$?-`m&y
z?4EH8q_y`iAwNEDtR-xbw?)&R@?qJ<##`)Y@G%exqx|t=vN_LYT!wm3R06^GA+4m6
z>XI;p8SY;VcOou++T_eL9vuxw9I1a&BF{%NtBHl4Wp2rq&xI>xzv;sDkc4u{kvipu
z+L;Zw(5K!Bj^&;WG(KhXH&8L}dzf2MrO!G3>Ad7vLQJlSM%p`QN#a}d4QYnK;br!9
z#$5XVISNX8LxN~ZKDm(gr{&v3Hm`Qhe|R6MYpfXtcv}i_?D*~5=bc#x?%sR|=KbC~
zBlV!>6J=lzXVzY&BgHFYtKM4?>?)0dB^6C|=Z51+4b%r!>5bDjcm||vSssjjVzkqf
z?SG<HYCg_;b4bKXFOPR~N09jfd%OI?Llt_QO9sSRoR!X;je~VJH)FhG98U(#=+Vg7
ztcy;AR2v6rv>IvbH8rp{Hl+L(uhFAfNymrHTvB_PSzj80mKg`?7sV1^vbs5rM%|Y0
xUl)1qG2(l^%3dy6yVR3xYxKL36OH|IwWWrm$mvlt68{e+N^<J5ucggG{|Aw1NE847

literal 0
HcmV?d00001

diff --git a/mo_gymnasium/envs/fruit_tree/assets/node.png b/mo_gymnasium/envs/fruit_tree/assets/node.png
new file mode 100644
index 0000000000000000000000000000000000000000..3a65643564a49059deaea03e1f69aeed53e5b587
GIT binary patch
literal 4001
zcmZ`+3p|tm_utmYHTN)DYi>&nBXSwWSaV;jC@E~rb=X{!ONdQlA;q`c$)#^3p`;X*
z+zFv95k~i1l2X5^@Bdq`|L^~Qp4aDd&U4OrpZ7WE^SoZq>tQ%K;6#NWLI40j6u%#P
zfG3%oOMsuZho=sa005wepQWV}-qI506h`&&3nT*oh_m7ByS5et=|^U{yIk0oSWL16
zI58#1a)1Q!+?S?u(DGw!zM%j4`E1H-Ez0hayRZq!QMf-v>=d)DvKJk)JNnY)a~nl8
z!&tLe!+8!(tRXzWX7-}_o#T*eU)=gXbyYtiK1e*$ClZwJ;gU$vgR8_Y_E%2Y-00|S
zKKV3cR<GqmIT8AF)mEwaP1}q0or8Hcvf+aU_`6p{q!dkGavQ4xIWq6|k`OPq{H1Ge
zdd+bB)phIM0~&}}%{QY4H*!8?*BN~2?T?*_y;>IEbKLR8C1i)J)mN3R4DHO+X0D_T
zNLfJ2>s=u<zErJqJ9CFfJ!z~XM*%dA2rElzmmbJ#+hf9T9HPEo|Ge!;!D7;aQuN}@
z*}q>{MhnEO9_CPpcdi4@&)euW)b<+r_f0<>CS|vD4TMtMv;;#~n2o-AkM3Ne+3D4-
z&6TZN;$=zq5iifY=MCJ8?27lXw+E>6bO8VeC<)lY(}27ikm?5j|I`5hRh|R@wq%_K
zY~@Kw-f}7e{p)2GZTYQl24b8o@pzte_6{SHgJ~2hJ-qFO1^@s;`?<K%UG43Rys1G*
zPZHIOjEo8j*@OU0qKtTI5Si`?iwX)1rWr+{;Xfmcc>1PU3l94kLJvU0UG1G<meepZ
zOdqL@)P|c1!C)|xFp`ha0j%{uaNZLdPNCC7jI^{OBO{TKx=3o6ua=IXp`n&GN(+V3
z<V9%GqJ!z4QJTRtrC&+@Cl8iP^A7V1q5DyTVVik9y{O@IG#tK3^zZenPI8prZ>C_{
zKW*_E)Y|lD=^(YW{x2Ba&*%SvZF+vee)jc?&SW#0k&|B(InWL37sQ(?uQXE~Lv53v
zH2>xKUFa_)kwy-)qz3VTbkpC*@(=h==l=%&q;&mFi9#9vN%@E859DSGMvh^AJT}kG
z8Jg;tX#JP%pLi3k%>n;0@L#9-)5<$eQy~+re_st#A^n*r4*&pR93G2ti2}a8)ZZ^N
zBL66w;GyErXCZp8j(ZwgyvJV_X~&V1-CHa%7~$Nr$KL&d>O-QGLAd9du$?SrH@-!u
z;<&{D*@r1d>k@{LOm<x;M!O^bl=uOO<hVQSc?-|K_FXy_ImfxF`$8?6x6k+F&BykC
zKG~OPINvjL3)nf?64{j1dy)&KtKHf%nQWSdEd5++Td8cyX^p()@aP#glH0^3F2iGh
z=p^W|!-v=Ny=ztY%MJ39rQvNgT#scuUHBF-$T|4zTKoWwE4ZA=#wFL&1%aI(h1TU`
zL|c>kj*yZ9&)V101#2+lptaY727@$iP{V7u0=Xv8!>satKF$Xi&4*4wf(*5Fok=a9
z>QG`LuT8Hn1xW)&;-_{k@1@fw&qOTW*)tX{cQReS_fp1$T;7OVi8(v9?t{QOsJZ}O
zXoaqx;nwczVIh4QXxK!U0&o$r4zCV=&BjiC@aC!~f1#d7ajH(sC`ZeNTrIw9o@y=m
z(IWaF=dPeD_pn--6qL^?jt_Ddw{rokKB+TpRB<~a^2^d~`}{owSf@AO!|q!rIOg<k
zUwp5<cN16|x&7wik&!L<I9!ERu_``vLAcZwczt?Iv*W@|+^*{oFEfVepheti_N!>r
z7d;NSqE^6XkpO$#P5OlL&l<kmP}ibZccswgfw8Z(G^!%8Tx?*g`UIC7AR&HwC)n=k
z>Bf-qmFm<UgmXM@a=TwK0DnMqvd~FpGEnWyiK<2$>HYPrN3j`;#JlBau-*etSrXv}
zhuLs&#XReB-J0;QvR^sZT2kiWF>ntKlyBshf{^>B$Y1SSDo@HChrzBm^1o4t_nN`p
zb87AQDz{@=Ja&f9MF->gJrt3|Y?{MS8(5Me--I&*FD85$58=vz^6Tpg^}BK2PXh||
zt#hV6y~4k%31U$LB@^G{-}tg;vsER7k`ZjI=%E{U_enj1dT)8LpD^SB8Xc77EX9{q
zDSU;)Qn>mg*V_1V0ON5ctlQ=2d{r6<uXkZpWf2)-obueI8sa;?&COq|%l;X6IIRXh
z<PzF=KBH?tnRvusijP_nX#mZw72W8eKHt%GWi%;(h&FMeCa#>S9o@L0v0`rzE1JD_
zw@hfg&fn{tFeHm&%79qM3A@X@4emH(OjaOO=@phcjFbk~;fGijwcaUR4!lo>x&$xt
zLV5w+MR3}|Iu4#g_M5RegjT-77|5Sl#M~JA5MflpKLfoiq`ud)g^3#sjLt55LO%9*
z-?{RxhMXL@5C{7@__+*{Z^6Rqz<D?F5kz@4RI<g|#^WMUVBW&I*<ocV3EnM?7E&HW
zIcAp$JboY=ubdMaye$kZBpCQAI9jU7KFrg7%{)kR6;VK*zP|>~49fxAWu?qxJo)v-
zXDA+57*Jl78+g4K-;!6m#tOB=&l|iX29wCU`%5WO3faoSH{v{;-D-MtRi2bi*<)cN
zhr073zjz&!80d3U_^6&wGlplIbnWuWQ&w^koyL-{)*n;KHeu=)KRM#n{Xx9>6wXao
zK8S@Yo^S&yWRs%64BBnO5SVFZMu3UXSqF&{x0T$N@DNDtrziAPT$&{0n*TslkOQRr
z<;G}!(JZ|*;`U;6yRtlV4=PLpw+pF^dPnG!m}cMwO_IKcLNiNhv#}@#X;$CO3Bf6m
zZy(TDJimK(b?WNlYCI~$|BTO%J8cZEzlC(4#}3xKuF{8i2W6tzn?dAbF$F~>wgoJ2
zSUOsIe+f{(RkKkpQ4b`xHCS1Jqo}O+5xH>TJDLLSU{vBIn+U7hAlrOR#S#nmXsa?G
zIeaRDM7VaaDsL4-hw54@+TG5y?E^K*Il7HLHEojg$~x0Nvm}=&I-OPxJ?Z(}3;aoZ
zOs#~gyS!bAo!SPKAp3fVq{;(BvDk4>#2IrS$QUc_ctXSc@x9mj3<PKAs2##$3M28L
z8^t^xwpO2(W#^y+wLT64eNIQj$?pU1IOZVkDoN0=E)oZx$gxOM1cKiX?2^I+I!h_f
z0@5WqEnyQJxYnumGv>(&XePsnKauAdJpVx8)vk~uP^;_DG^}A_iSMz}Cux0~)vX=G
z-prS6_p~-wlpr7>LP|~vZB`Pq78vEN3W>t-F8P_`U?K5+*QIj<zO}$gl2Tc1Vy%F{
zF)$+g>j~jD3%iqk&%D6(;%Uvr`b_MJCbu)@+d&crUL_YhnJ)vZ0bqD=N07AUI6M7g
z!+x3)E1{w&(DUs2NV42=+_%X#J?wW)uyxa%(1mYL#^e%pz>*HU@RT2(8d;7k^ZSgl
zMF~|GGY)&~RhmR6hrKU(z12}9J5$Mh{~`3LIrNM-7z9#sm#dAzD)vS##jUl^*?#wR
z<!ZeyP5L3HZ{63rPrR9(pyRQEm}<gT#*4@#?p2hlit%#cZcuU}XtD2Rio_A0KRyOL
zeu=@%^i)W6YJ9~*t%zA>>`SMqa(740?*5VZW`O;LA~&}0{BCeYrh(T?on2;$`NZyZ
zjr82GD4Z&AD1NZ=n`&B^(w-W()&hh_^GnxJ`#Yny;0${>Bg}ZP6d_()f41GA@^hCg
zY~$P!(py4WNrYd8-7<o?Be*A!&$nVOJ)PP1(j*Jvd=skDx3}OycD|SPdZeAI+ALY~
zX=zz-EPCac+1C{Fs(Q^?Lb9;O>@mfN&%&km=I1nseVN@wPAQ1Yvr{U=le4JK2(ye^
z?U|N2wOtnuH!c#|4+`FKRwX=g6998|+e*$>G~gK59c0S*FyCFtBAs5)e#<PinhPGO
zh{Idx@+XVBk`$YC1)g_g1&LdgzAG=Mk4RmXgN7u~-a9HJ)$hZd8&9lOCakJUSs8pR
z5;MGA`yw~RVrs)F;vDO$`$}2i6p{vE_5DCj%Jg+TZom?ZXKm@W2dD3h%7gX1LybAu
zx7mje)GRUzrJ=XeX;cg(X{fcd%jc<{G9kAD&XjBetRxg*_9l2QYzL`l`&!7`lgEd#
zgHLo`>7~vS{@!|xz6E&h!BTNxMPsEzNvDv=dUCsM<tV?~TzW#+RJoHX%XLVOKoULe
zSMWA&xWFo*2ba9@-Dx|^O`BNjn<Yd4t5E4pZRIhT1E}iKYya%ZTW2GwE{At9HIg+P
zL)R1*#X^MHt5&TFd)-Go0^JWQF*Q;&p6!dxh`DH$(4YoUBgnd<2M(4E>UTA#-23p>
z-OLy|xAah4yb)p-z*4SnAeL}}q$$;EH<t-&yOt!MjV@f{g;eAH<~(J!%R~Mny5C)^
zOU~b1RU3Aeet54c0_nxotn=vW!}{1r&kEF2OuO;J*@fW!;fDR1j+R+AOZaLtpXH3v
znu5VcR>VOfs%o4_ErqV`+)DX$vo7YySZ}^E%5M&;!rH>Il4>-wLbsTS_DsC+uQL_K
z=uxAQC|YJ#UpXqHy7&a3wP4%QdjX(lNn;x13QdWjkynFb6z$)**M7TuA%HvZ)F?o+
zE;stSDqp{Qd;}9=0jj5Xej_Yf(J8fn;cbK4Bx~5D=~JQieUX=xY|a3ZqtjmC24n|j
z*J2zqCl>Z=xrkY?dp7WMm)`eDs_^sQuiOzXKs-0xBq~tLR(res<96`JWsw957S<~i
zBh<=5R(D3sdX`H#UF;-*0nzAn)tGHZ!9$(a<LZCSPdvGJW)WZkh?m(AX+6!s%xl(T
z8apdSOr<gN&k#gR#qnk46zrDlNC3z_l+x1j_q!_|D*;g(;Fx6nM|P=W(LOap#wDa;
zzUaoVlRIy1on*W-w?f!gAIP+s{2-UUCU<vXlrVX4#ZJ#~Ke#!~!@R#}B5Wxjqc!&j
Z5b-0YmT@$L_iqLO;H@06S1pd8`VTK3)Xo3^

literal 0
HcmV?d00001

diff --git a/mo_gymnasium/envs/fruit_tree/fruit_tree.py b/mo_gymnasium/envs/fruit_tree/fruit_tree.py
index fac03c06..e1c7aee6 100644
--- a/mo_gymnasium/envs/fruit_tree/fruit_tree.py
+++ b/mo_gymnasium/envs/fruit_tree/fruit_tree.py
@@ -5,6 +5,9 @@
 import numpy as np
 from gymnasium import spaces
 from gymnasium.utils import EzPickle
+import pygame
+from os import path
+from typing import Optional
 
 
 FRUITS = {
@@ -264,17 +267,21 @@ class FruitTreeEnv(gym.Env, EzPickle):
     The episode terminates when the agent reaches a leaf node.
     """
 
-    def __init__(self, depth=6):
+    metadata = {"render_modes": ["human", "rgb_array"], "render_fps": 1}
+    window_padding = 15
+    font_size = 12
+
+    def __init__(self, depth=6, render_mode: Optional[str] = None):
         assert depth in [5, 6, 7], "Depth must be 5, 6 or 7."
         EzPickle.__init__(self, depth)
 
+        self.render_mode = render_mode
         self.reward_dim = 6
         self.tree_depth = depth  # zero based depth
         branches = np.zeros((int(2**self.tree_depth - 1), self.reward_dim))
+        fruits = np.array(FRUITS[str(depth)])
         # fruits = np.random.randn(2**self.tree_depth, self.reward_dim)
         # fruits = np.abs(fruits) / np.linalg.norm(fruits, 2, 1, True)
-        # print(fruits*10)
-        fruits = np.array(FRUITS[str(depth)])
         self.tree = np.concatenate([branches, fruits])
 
         self.max_reward = 10.0
@@ -288,8 +295,24 @@ def __init__(self, depth=6):
         self.current_state = np.array([0, 0], dtype=np.int32)
         self.terminal = False
 
+        # pygame
+        self.row_height = 20
+        self.window_size = (1200, self.row_height * self.tree_depth + 150)
+        self.pix_square_size = np.array([10, 10], dtype=np.int32)
+        
+
+        self.window = None
+        self.clock = None
+        self.node_img = None
+        self.agent_img = None
+
     def get_ind(self, pos):
         return int(2 ** pos[0] - 1) + pos[1]
+    
+    def ind_to_state(self, ind):
+        x = int(np.log2(ind + 1))
+        y = ind - 2 ** x + 1
+        return np.array([x, y], dtype=np.int32)
 
     def get_tree_value(self, pos):
         return np.array(self.tree[self.get_ind(pos)], dtype=np.float32)
@@ -325,5 +348,100 @@ def step(self, action):
         reward = self.get_tree_value(self.current_state)
         if self.current_state[0] == self.tree_depth:
             self.terminal = True
-
+        
         return self.current_state.copy(), reward, self.terminal, False, {}
+    
+    def get_pos_in_window(self, row, index_in_row):
+        pos_x = self.window_padding + (index_in_row + 0.5) * (self.window_size[0] - 2 * self.window_padding) / (2 ** (row))
+        pos_y = row * self.row_height
+        return np.array([pos_x, pos_y])
+
+    def render(self):
+        if self.render_mode is None:
+            assert self.spec is not None
+            gym.logger.warn(
+                "You are calling render method without specifying any render mode. "
+                "You can specify the render_mode at initialization, "
+                f'e.g. mo_gym.make("{self.spec.id}", render_mode="rgb_array")'
+            )
+            return
+
+        if self.window is None:
+            pygame.init()
+
+            if self.render_mode == "human":
+                pygame.display.init()
+                pygame.display.set_caption("Fruit Tree")
+                self.window = pygame.display.set_mode(self.window_size)
+            else:
+                self.window = pygame.Surface(self.window_size)
+
+            if self.clock is None:
+                self.clock = pygame.time.Clock()
+
+            if self.node_img is None:
+                filename = path.join(path.dirname(__file__), "assets", "node.png")
+                self.node_img = pygame.transform.scale(pygame.image.load(filename), self.pix_square_size)
+                self.node_img = pygame.transform.flip(self.node_img, flip_x=True, flip_y=False)
+            if self.agent_img is None:
+                filename = path.join(path.dirname(__file__), "assets", "agent.png")
+                self.agent_img = pygame.transform.scale(pygame.image.load(filename), self.pix_square_size)
+                
+            # self.font = pygame.font.Font(path.join(path.dirname(__file__), "assets", "Minecraft.ttf"), 20)
+            self.font = pygame.font.Font(None, self.font_size)
+
+        canvas = pygame.Surface(self.window_size)
+        canvas.fill((0, 0, 0))
+
+        self.window.blit(canvas, (0, 0))
+
+        for ind, node in enumerate(self.tree):
+            row, index_in_row = self.ind_to_state(ind)
+            
+            node_pos = self.get_pos_in_window(row, index_in_row)
+
+            # Get childerns' positions for drawing branches
+            child1_pos = self.get_pos_in_window(row + 1, 2 * index_in_row)
+            child2_pos = self.get_pos_in_window(row + 1, 2 * index_in_row + 1)
+
+            half_square = self.pix_square_size / 2
+            
+            if (row, index_in_row) == tuple(self.current_state):
+                img = self.agent_img
+                font_color = (255, 0, 0) # Red digits for agent node
+            else:
+                img = self.node_img
+                font_color = (0, 255, 0) # Green digits for non-agent nodes
+
+            if row != self.tree_depth:
+                pygame.draw.line(self.window, (255,255,255), node_pos + half_square, child1_pos + half_square, 1)
+                pygame.draw.line(self.window, (255,255,255), node_pos + half_square, child2_pos + half_square, 1)
+                
+            
+            self.window.blit(img, np.array(node_pos))
+            if row == self.tree_depth:
+                values_imgs = [self.font.render(f'{val:.2f}', True, font_color) for val in node]
+                for i, val_img in enumerate(values_imgs):
+                    self.window.blit(val_img, node_pos + np.array([- 5, (i+1)*self.font_size]))
+
+        if self.render_mode == "human":
+            pygame.event.pump()
+            pygame.display.update()
+            self.clock.tick(self.metadata["render_fps"])
+        elif self.render_mode == "rgb_array":
+            return np.transpose(np.array(pygame.surfarray.pixels3d(self.window)), axes=(1, 0, 2))
+        
+if __name__ == "__main__":
+    import mo_gymnasium as mo_gym
+    import time
+
+    env = mo_gym.make("fruit-tree", depth=6, render_mode="human")
+    terminated = False
+    env.reset()
+    while True:
+        env.render()
+        obs, r, terminated, truncated, info = env.step(env.action_space.sample())
+        if terminated or truncated:
+            env.render()
+            time.sleep(2)
+            env.reset()