nesticot commited on
Commit
a9d3e04
·
verified ·
1 Parent(s): 673cf15

Update stuff_model/calculate_arm_angles.py

Browse files
stuff_model/calculate_arm_angles.py CHANGED
@@ -93,9 +93,9 @@ def calculate_arm_angles(df: pl.DataFrame, pitcher_id: int) -> pl.DataFrame:
93
 
94
  # Adjust based on data source freshness
95
  if old_data:
96
- df_filter = df_filter.with_columns(((pl.col("arm_angle") * 0.5) + (ball_angle * 0.5)).alias("arm_angle"))
97
  elif daily_check:
98
- df_filter = df_filter.with_columns(((pl.col("arm_angle") * 0.25) + (ball_angle * 0.75)).alias("arm_angle"))
99
  else:
100
  df_filter = df_filter.with_columns(((pl.col("arm_angle") * 0.0) + (ball_angle * 1)).alias("arm_angle"))
101
 
 
93
 
94
  # Adjust based on data source freshness
95
  if old_data:
96
+ df_filter = df_filter.with_columns(((pl.col("arm_angle") * 1) + (ball_angle * 0.0)).alias("arm_angle"))
97
  elif daily_check:
98
+ df_filter = df_filter.with_columns(((pl.col("arm_angle") * 1.0) + (ball_angle * 0.0)).alias("arm_angle"))
99
  else:
100
  df_filter = df_filter.with_columns(((pl.col("arm_angle") * 0.0) + (ball_angle * 1)).alias("arm_angle"))
101