add another comment

rapidsai · Aug 17, 2021 · 4192518 · 4192518
1 parent 1e7a55c
commit 4192518
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/cugraph/utilities/nx_factory.py b/python/cugraph/utilities/nx_factory.py
@@ -37,6 +37,7 @@ def convert_from_nx(nxG, weight=None):
 
     pdf = nx.to_pandas_edgelist(nxG)
     # Convert vertex columns to strings if they are not integers
+    # This allows support for any vertex input type
     if pdf["source"].dtype not in [np.int32, np.int64] or \
             pdf["target"].dtype not in [np.int32, np.int64]:
         pdf['source'] = pdf['source'].astype(str)