hG SSKJrJr SSKrSSKrSSKJr SSK J r Sr Sr Sr\ \ \/rSrS \R"S \S \R"4S jrS \R"S\R"S \S\S \\R"\R"44 SjrS\R,S\S \\R"\R"44SjrSS\\\R"\R"4S\S\S\\\4S S4 SjjrSS\\S\\S\S\S\\\4S S4 Sjjrg))CallableOptionalN)pyplot) load_results timestepsepisodes walltime_hrsdarraywindowreturncURSSURSU- S-U4-n/URQURSP7n[RRR XUS9$)z Apply a rolling window to a np.ndarray :param array: the input Array :param window: length of the rolling window :return: rolling window on the input array N)shapestrides)rrnplib stride_tricks as_strided)r r rrs b/home/james-whalen/.local/lib/python3.13/site-packages/stable_baselines3/common/results_plotter.pyrolling_windowrsl KK  B& 81 plot_curves..Us,Gb1b Gsr)szEpisode RewardsN)pltfiguremax enumeratescatterrEPISODES_WINDOWr!rmeanplotxlimr4xlabelylabel tight_layout) r3r#r4r5max_xmin_x_xyy_means r plot_curvesrQGsJJu& ,G, ,E Ew' 6A AA 771: (#A/277CIA HHQ  (HHUIIeJJvJJ !rdirs num_timesteps task_namec/nUHAn[U5nUbXwRR5U:*nURU5 MC UVs/sHn[ Xr5PM nn[ XX45 gs snf)a Plot the results using csv files from ``Monitor`` wrapper. :param dirs: the save location of the results to plot :param num_timesteps: only plot the points below this value :param x_axis: the axis for the x and y output (can be X_TIMESTEPS='timesteps', X_EPISODES='episodes' or X_WALLTIME='walltime_hrs') :param task_name: the title of the task to plot :param figsize: Size of the figure (width, height) N)rr'r&appendr2rQ) rRrSr#rTr5 data_framesfolderr"r3s r plot_resultsrYeswK!&)  $#LL$7$7$9]$JKJ:&  rhs|%%9    J7 P"** Pc Pbjj P 1rzz 1"** 1c 1 1V[\^\f\fhjhrhr\rVs 1bllCE"**bjj:P4Q2gm % BJJ./ 0:=FITYZ]_bZbTc >lr5 s)5$,SM5;>5KN5Y^_bdg_gYh5 5r