a81d27dc67a9c14435b3bd9bf7122fa0871bf76e,python/dgl/cgraph.py,DGLGraph,in_edges,#DGLGraph#Any#,70

Before Change

        if isinstance(v, int):
            edge_array = _CAPI_DGLGraphInEdges_1(self._handle, v)
        else:
            v = utils.Index(v)
            v_array = F.asdglarray(v.totensor())
            edge_array = _CAPI_DGLGraphInEdges_2(self._handle, v_array)
        src = edge_array(0)
        dst = edge_array(1)

After Change

        utils.Index
            The edge ids.
        
        if len(v) == 1:
            edge_array = _CAPI_DGLGraphInEdges_1(self._handle, v[0])
        else:
            v_array = v.todgltensor()

Instances