Update models/pitcher_baseline_store.py

models/pitcher_baseline_store.py

@@ -168,6 +168,8 @@ def load_pitcher_cross_game_baseline(
                     AVG(extension) AS stored_baseline_extension,
                     AVG(pfx_x) AS stored_baseline_pfx_x,
                     AVG(pfx_z) AS stored_baseline_pfx_z,
+                    COUNT(pfx_x) AS stored_baseline_pfx_x_sample_size,
+                    COUNT(pfx_z) AS stored_baseline_pfx_z_sample_size,
                     COUNT(velocity) AS stored_baseline_velocity_sample_size,
                     COUNT(spin_rate) AS stored_baseline_spin_sample_size,
                     COUNT(extension) AS stored_baseline_extension_sample_size
@@ -192,6 +194,8 @@ def load_pitcher_cross_game_baseline(
                         AVG(extension) AS stored_pitch_type_baseline_extension,
                         AVG(pfx_x) AS stored_pitch_type_baseline_pfx_x,
                         AVG(pfx_z) AS stored_pitch_type_baseline_pfx_z,
+                        COUNT(pfx_x) AS stored_baseline_pfx_x_sample_size,
+                        COUNT(pfx_z) AS stored_baseline_pfx_z_sample_size,
                         COUNT(velocity) AS stored_pitch_type_baseline_velocity_sample_size,
                         COUNT(spin_rate) AS stored_pitch_type_baseline_spin_sample_size,
                         COUNT(extension) AS stored_pitch_type_baseline_extension_sample_size