Update stuff_model/calculate_arm_angles.py

stuff_model/calculate_arm_angles.py

@@ -105,8 +105,11 @@ def calculate_arm_angles(df: pl.DataFrame,pitcher_id:int) -> pl.DataFrame:
 
         
         df_filter = df_filter.with_columns(
-            df_filter["arm_angle"].fill_nan(df_filter["arm_angle"].mean())
-        )        
+            df_filter["arm_angle"]
+            .fill_nan(None)  # Convert NaN to null
+            .fill_null(df_filter["arm_angle"].mean())  # Fill nulls with mean
+        )
+   
 
         print([x for x in df_filter["arm_angle"]])