diff --git a/sbdataextraction/sbdataextraction.py b/sbdataextraction/sbdataextraction.py index 5833a38..14d56c4 100644 --- a/sbdataextraction/sbdataextraction.py +++ b/sbdataextraction/sbdataextraction.py @@ -210,7 +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(): + is_pass = (event_name == "pass") + if is_pass and "assisted_shot_id" in events["pass"].keys(): # noqa related.append(events["pass"]["assisted_shot_id"]) else: related = None