code: def assert_same_edge_feats(gs, g):
      set(gs.edata.keys()) == set(g.edata.keys())
      for key in g.edata.keys():
          for etype in g.canonical_etypes:
              indices = th.arange(
                  0, g.num_edges(etype), dtype=g.idtype
              ).cuda()
              if len(g.etypes) <= 1 or etype in g.edata[key]:
                  g_output = g.get_edge_storage(
                      key=key, etype=etype
                  ).fetch(indices, device="cuda")
                  gs_output = gs.get_edge_storage(
                      key=key, etype=etype
                  ).fetch(indices)
                  equal_t = (gs_output != g_output).sum().cpu()
                  assert equal_t == 0
  
imports: from cugraph.utilities.utils import import_optional
label_desc: Use image node_id 5 for calling a global function with example usage: assert_same_edge_feats(gs, g) without return types