import numpy as np

def load_graph(config):
    dataset_path = config['data']['graph_pkl_filename']
    graph = np.load(dataset_path)
    # 将inf值填充为0
    graph = np.nan_to_num(graph, nan=0.0, posinf=0.0, neginf=0.0)
    
    return graph