updates based on PR review

Author: BradReesWork

python/cugraph/cugraph/components/connectivity.py

@@ -31,9 +31,8 @@ def _ensure_args(api_name, G, directed, connection, return_labels):
     args with proper defaults if not specified, or raises TypeError or
     ValueError if incorrectly specified.
     """
-    G_type = type(G)
     # Check for Graph-type inputs and set defaults if unset
-    if G_type == Graph:
+    if isinstance(G, Graph):
         exc_value = "'%s' cannot be specified for a Graph-type input"
         if directed is not None:
             raise TypeError(exc_value % "directed")

python/cugraph/cugraph/link_prediction/cosine.py

@@ -102,7 +102,7 @@ def cosine(
 
     if isinstance(vertex_pair, cudf.DataFrame):
         vertex_pair = renumber_vertex_pair(input_graph, vertex_pair)
-        vertex_pair = ensure_valid_dtype(input_graph, vertex_pair, "Cosine")
+        vertex_pair = ensure_valid_dtype(input_graph, vertex_pair)
         src_col_name = vertex_pair.columns[0]
         dst_col_name = vertex_pair.columns[1]
         first = vertex_pair[src_col_name]
@@ -186,9 +186,9 @@ def cosine_coefficient(
 
     """
     warnings.warn(
-        "deprecated as of 25.10. This function was to match "
-        "NetworkX which is no longer needed, use networkx directly with"
-        "the ``nx-cugraph`` backend. See:  https://rapids.ai/nx-cugraph/",
+        "deprecated as of 25.10. Use `cosine()` instead. "
+        "If calling with a NetworkX Graph object, use networkx with the "
+        "nx-cugraph backend. See: https://rapids.ai/nx-cugraph",
         DeprecationWarning,
     )
 

python/cugraph/cugraph/link_prediction/jaccard.py

@@ -105,7 +105,7 @@ def jaccard(
 
     if isinstance(vertex_pair, cudf.DataFrame):
         vertex_pair = renumber_vertex_pair(input_graph, vertex_pair)
-        vertex_pair = ensure_valid_dtype(input_graph, vertex_pair, "Jaccard")
+        vertex_pair = ensure_valid_dtype(input_graph, vertex_pair)
         src_col_name = vertex_pair.columns[0]
         dst_col_name = vertex_pair.columns[1]
         first = vertex_pair[src_col_name]
@@ -191,9 +191,9 @@ def jaccard_coefficient(
 
     """
     warnings.warn(
-        "deprecated as of 25.10. This function was to match "
-        "NetworkX which is no longer needed, use networkx directly with"
-        "the ``nx-cugraph`` backend. See:  https://rapids.ai/nx-cugraph/",
+        "deprecated as of 25.10. Use `jaccard()` instead. "
+        "If calling with a NetworkX Graph object, use networkx with the "
+        "nx-cugraph backend. See: https://rapids.ai/nx-cugraph",
         DeprecationWarning,
     )
 

python/cugraph/cugraph/link_prediction/overlap.py

@@ -86,9 +86,9 @@ def overlap_coefficient(
     >>> df = overlap_coefficient(G)
     """
     warnings.warn(
-        "deprecated as of 25.10. This function was to match "
-        "NetworkX which is no longer needed, use networkx directly with"
-        "the ``nx-cugraph`` backend. See:  https://rapids.ai/nx-cugraph/",
+        "deprecated as of 25.10. Use `overlap()` instead. "
+        "If calling with a NetworkX Graph object, use networkx with the "
+        "nx-cugraph backend. See: https://rapids.ai/nx-cugraph",
         DeprecationWarning,
     )
 
@@ -185,7 +185,7 @@ def overlap(
 
     if isinstance(vertex_pair, cudf.DataFrame):
         vertex_pair = renumber_vertex_pair(input_graph, vertex_pair)
-        vertex_pair = ensure_valid_dtype(input_graph, vertex_pair, "Overlap")
+        vertex_pair = ensure_valid_dtype(input_graph, vertex_pair)
         src_col_name = vertex_pair.columns[0]
         dst_col_name = vertex_pair.columns[1]
         first = vertex_pair[src_col_name]

python/cugraph/cugraph/link_prediction/sorensen.py

@@ -104,7 +104,7 @@ def sorensen(
 
     if isinstance(vertex_pair, cudf.DataFrame):
         vertex_pair = renumber_vertex_pair(input_graph, vertex_pair)
-        vertex_pair = ensure_valid_dtype(input_graph, vertex_pair, "Sorensen")
+        vertex_pair = ensure_valid_dtype(input_graph, vertex_pair)
         src_col_name = vertex_pair.columns[0]
         dst_col_name = vertex_pair.columns[1]
         first = vertex_pair[src_col_name]
@@ -193,9 +193,9 @@ def sorensen_coefficient(
 
     """
     warnings.warn(
-        "deprecated as of 25.10. This function was to match "
-        "NetworkX which is no longer needed, use networkx directly with"
-        "the ``nx-cugraph`` backend. See:  https://rapids.ai/nx-cugraph/",
+        "deprecated as of 25.10. Use `sorensen()` instead. "
+        "If calling with a NetworkX Graph object, use networkx with the "
+        "nx-cugraph backend. See: https://rapids.ai/nx-cugraph",
         DeprecationWarning,
     )
 

python/cugraph/cugraph/traversal/bfs.py

@@ -40,10 +40,8 @@ def _ensure_args(G, start, i_start, directed):
 
     start = start if start is not None else i_start
 
-    G_type = type(G)
-
     # Check for Graph-type inputs
-    if G_type is Graph:
+    if isinstance(G, Graph):
         if directed is not None:
             raise TypeError("'directed' cannot be specified for a " "Graph-type input")
 

python/cugraph/cugraph/traversal/sssp.py

@@ -285,9 +285,9 @@ def shortest_path(
     for details.
     """
     warnings.warn(
-        "deprecated as of 25.10. This function was to match "
-        "NetworkX which is no longer needed, use networkx directly with"
-        "the ``nx-cugraph`` backend. See:  https://rapids.ai/nx-cugraph/",
+        "deprecated as of 25.10. Use `sssp()` instead. "
+        "If calling with a NetworkX Graph object, use networkx with the "
+        "nx-cugraph backend. See: https://rapids.ai/nx-cugraph",
         DeprecationWarning,
     )
 

python/cugraph/cugraph/utilities/utils.py

@@ -195,7 +195,10 @@ def get_traversed_path_list(df, id):
     return answer
 
 
-def ensure_valid_dtype(input_graph, vertex_pair, func_name):
+def ensure_valid_dtype(input_graph, vertex_pair):
+    import inspect
+    func_name = inspect.stack()[1].function
+
     vertex_dtype = input_graph.edgelist.edgelist_df.dtypes.iloc[0]
     vertex_pair_dtypes = vertex_pair.dtypes