Spaces:
Sleeping
Sleeping
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") *
|
| 97 |
elif daily_check:
|
| 98 |
-
df_filter = df_filter.with_columns(((pl.col("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 |
|