Update models/pitcher_adjustment.py

models/pitcher_adjustment.py

@@ -20,6 +20,10 @@ def build_pitcher_feature_row(statcast_df: pd.DataFrame, pitcher_name: str) -> d
             "hard_hit_rate_allowed": None,
             "barrel_rate_allowed": None,
             "avg_release_speed": None,
+            "avg_release_spin_rate": None,
+            "avg_release_extension": None,
+            "avg_pfx_x": None,
+            "avg_pfx_z": None,
             "sample_size": 0,
         }
 
@@ -31,15 +35,27 @@ def build_pitcher_feature_row(statcast_df: pd.DataFrame, pitcher_name: str) -> d
             "hard_hit_rate_allowed": None,
             "barrel_rate_allowed": None,
             "avg_release_speed": None,
+            "avg_release_spin_rate": None,
+            "avg_release_extension": None,
+            "avg_pfx_x": None,
+            "avg_pfx_z": None,
             "sample_size": 0,
         }
 
     launch_speed = pd.to_numeric(df.get("launch_speed"), errors="coerce")
     launch_angle = pd.to_numeric(df.get("launch_angle"), errors="coerce")
     release_speed = pd.to_numeric(df.get("release_speed"), errors="coerce")
+    release_spin_rate = pd.to_numeric(df.get("release_spin_rate"), errors="coerce")
+    release_extension = pd.to_numeric(df.get("release_extension"), errors="coerce")
+    pfx_x = pd.to_numeric(df.get("pfx_x"), errors="coerce")
+    pfx_z = pd.to_numeric(df.get("pfx_z"), errors="coerce")
 
     ev_allowed = _safe_mean(launch_speed)
     avg_release_speed = _safe_mean(release_speed)
+    avg_release_spin_rate = _safe_mean(release_spin_rate)
+    avg_release_extension = _safe_mean(release_extension)
+    avg_pfx_x = _safe_mean(pfx_x)
+    avg_pfx_z = _safe_mean(pfx_z)
 
     hard_hit_rate_allowed = None
     if not launch_speed.dropna().empty:
@@ -66,6 +82,10 @@ def build_pitcher_feature_row(statcast_df: pd.DataFrame, pitcher_name: str) -> d
         "hard_hit_rate_allowed": hard_hit_rate_allowed,
         "barrel_rate_allowed": barrel_rate_allowed,
         "avg_release_speed": avg_release_speed,
+        "avg_release_spin_rate": avg_release_spin_rate,
+        "avg_release_extension": avg_release_extension,
+        "avg_pfx_x": avg_pfx_x,
+        "avg_pfx_z": avg_pfx_z,
         "sample_size": int(len(df)),
     }