Update functions/pitch_summary_functions.py

functions/pitch_summary_functions.py

@@ -338,7 +338,7 @@ def tj_stuff_roling_game(df: pl.DataFrame, window: int, ax: plt.Axes):
         ])
 
         sns.lineplot(x=range(1, max(df_item['start_number_right']) + 1),
-                        y=df_item.filter(pl.col('pitch_type') == i)['tj_stuff_plus'].rolling_mean(window),
+                        y=df_item.filter(pl.col('pitch_type') == i)['tj_stuff_plus'].rolling_mean(window,min_periods=1),
                         color=dict_colour[i],
                         ax=ax, linewidth=3)
 
@@ -346,14 +346,14 @@ def tj_stuff_roling_game(df: pl.DataFrame, window: int, ax: plt.Axes):
         for n in range(len(df_item)):
             if df_item['game_id'].is_null()[n]:
                 sns.scatterplot(x=[df_item['start_number_right'][n]],
-                                y=[df_item['tj_stuff_plus'].rolling_mean(window)[n]],
+                                y=[df_item['tj_stuff_plus'].rolling_mean(window,min_periods=1)[n]],
                                 color='white',
-                                ec='black',
+                                ec=dict_colour[i],
                                 ax=ax,
                                 zorder=100)
 
     # Adjust x-axis limits to start from 1
-    ax.set_xlim(window, max(df_item['start_number']))
+    ax.set_xlim(1, max(df_item['start_number']))
     ax.set_ylim(70, 130)
     ax.set_xlabel('Games', fontdict=font_properties_axes)
     ax.set_ylabel('tjStuff+', fontdict=font_properties_axes)