
    h'                     B    S SK rS\R                  R                  4S jrg)    N	np_randomc                     [         R                  " U 5      n [         R                  " U 5      n[         R                  " X!R	                  5       :  5      $ )zRSample from categorical distribution where each row specifies class probabilities.)npasarraycumsumargmaxrandom)prob_nr   csprob_ns      W/home/james-whalen/.local/lib/python3.13/site-packages/gymnasium/envs/toy_text/utils.pycategorical_sampler      s9    ZZFyy H99X 0 0 2233    )numpyr   r	   	Generatorr    r   r   <module>r      s    4"))*=*= 4r   