Update stuff_model/calculate_arm_angles.py

stuff_model/calculate_arm_angles.py

@@ -8,10 +8,13 @@ def calculate_arm_angles(df: pl.DataFrame, pitcher_id: int) -> pl.DataFrame:
         headers = {
             'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3'
         }
-        response = requests.get(url, headers=headers)
-        if not response.ok or "<html" in response.text.lower():
+        try:
+            response = requests.get(url, headers=headers, timeout=5)
+            if not response.ok or "<html" in response.text.lower():
+                return None
+            return pl.read_csv(StringIO(response.text), truncate_ragged_lines=True)
+        except requests.RequestException:
             return None
-        return pl.read_csv(StringIO(response.text), truncate_ragged_lines=True)
 
     date_start = df['game_date'][0]
     date_end = df['game_date'][-1]