Added assisted shots to related events for passes

RobBlumberg · Apr 12, 2020 · 1d83de6 · 1d83de6
1 parent 9531974
commit 1d83de6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sbdataextraction/sbdataextraction.py b/sbdataextraction/sbdataextraction.py
@@ -210,6 +210,8 @@ def get_events_for_game(self):
 
                 if "related_events" in events.keys():
                     related = events["related_events"]
+                    if event_name == "pass" and "assisted_shot_id" in events["pass"].keys():
+                        related.append(events["pass"]["assisted_shot_id"])
                 else:
                     related = None