def get_game_states(replay_file): events = get_events(replay_file) game_data = GameData() game_data.record_events(events) return game_data.normalized_states
def get_nparray(replay_file): events = get_events(replay_file) game_data = GameData() game_data.record_events(events) return game_data.convert_to_nparray()