Update stuff_model/calculate_arm_angles.py

stuff_model/calculate_arm_angles.py

@@ -103,15 +103,17 @@ def calculate_arm_angles(df: pl.DataFrame,pitcher_id:int) -> pl.DataFrame:
                     ((pl.col("arm_angle") * 0.0) + (ball_angle * 1)).alias("arm_angle")
                 )        
 
+        valid_mean = df_filter["arm_angle"].fill_nan(None).drop_nulls().mean()
+
         
         df_filter = df_filter.with_columns(
             df_filter["arm_angle"]
             .fill_nan(None)  # Convert NaN to null
-            .fill_null(df_filter["arm_angle"].mean())  # Fill nulls with mean
+            .fill_null(valid_mean)  # Fill nulls with mean
         )
    
 
-        print([x for x in df_filter["arm_angle"]])
+        #print([x for x in df_filter["arm_angle"]])