def heuristics_v1(edge_attr: np.ndarray) -> np.ndarray:
    return np.ones(edge_attr.shape[0])
