fixes

rapidsai · Nov 2, 2023 · e04ca18 · e04ca18
1 parent f4bcdc2
commit e04ca18
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/python/cugraph/cugraph/structure/graph_classes.py b/python/cugraph/cugraph/structure/graph_classes.py
@@ -469,8 +469,7 @@ def from_numpy_array(self, np_array, nodes=None):
         nodes: array-like or None, optional (default=None)
             A list of column names, acting as labels for nodes
         """
-        if not isinstance(np_array, np.ndarray):
-            raise TypeError("np_array input is not a Numpy array")
+        np_array = np.asarray(np_array)
         if len(np_array.shape) != 2:
             raise ValueError("np_array is not a 2D matrix")
 

diff --git a/python/cugraph/cugraph/structure/graph_implementation/simpleGraph.py b/python/cugraph/cugraph/structure/graph_implementation/simpleGraph.py
@@ -1288,7 +1288,7 @@ def nodes(self):
             else:
                 return cudf.concat(
                     [df[simpleGraphImpl.srcCol], df[simpleGraphImpl.dstCol]]
-                ).unique()
+                ).drop_duplicates()
         if self.adjlist is not None:
             return cudf.Series(np.arange(0, self.number_of_nodes()))