Update app

data.py

@@ -9,7 +9,7 @@ import os
 import re
 
 from translate import (
-    translate_pa_outcome, translate_pitch_outcome, 
+    translate_pa_outcome, translate_pitch_outcome,
     jp_pitch_to_en_pitch, jp_pitch_to_pitch_code,
     max_pitch_types
 )

gradio_function.py

@@ -53,7 +53,7 @@ colorscale = [
 ]
 
 
-def plot_pitch_map(df, player=None, loc=None, pitch_type=None, pitch_name=None, all_pitches=False, min_pitches=2):
+def plot_pitch_map(df, player=None, loc=None, pitch_type=None, pitch_name=None, all_pitches=False, min_pitches=3):
   assert not ((loc is None and player is None) or (loc is not None and player is not None)), 'exactly one of `player` or `loc` must be specified'
 
   if loc is None and player is not None:
@@ -124,9 +124,6 @@ def plot_pitch_velo(df=None, player=None, velos=None, pitch_type=None, pitch_nam
     # velos = df.set_index(['name', pitch_col]).sort_index().loc[(player, pitch_val), 'release_speed']
     velos = df.filter((pl.col('name') == player) & (pl.col(pitch_col) == pitch_val))['release_speed']
 
-  if isinstance(velos, int):
-    velos = [velos] # is this line still necessary after porting to polars?
-
   fig = go.Figure()
   if len(velos) >= min_pitches:
     fig = fig.add_trace(go.Violin(x=velos, side='positive', hoveron='points', points=False, meanline_visible=True, name='Velocity Distribution'))
@@ -312,7 +309,7 @@ def get_data(player, handedness):
 
     pitch_velos.append(gr.update(
         # value=plot_pitch_velo(velos=_df_by_pitch_name.loc[pitch_name, 'release_speed']),
-        value=plot_pitch_velo(velos=_df.filter(pl.col('pitch_name') == pitch_name)['release_speed']),
+        value=plot_pitch_velo(velos=_df.filter((pl.col('pitch_name') == pitch_name) & (pl.col('release_speed').is_not_null()))['release_speed']),
         visible=True
     ))
     pitch_maps.append(gr.update(