From 06da8e23c87d4b9dbe97ac6fb81dd2d1262345a5 Mon Sep 17 00:00:00 2001
From: Matthew Roeschke <10647082+mroeschke@users.noreply.github.com>
Date: Mon, 6 May 2024 14:32:16 -0700
Subject: [PATCH] Address k_truss_subgraph(use_weights=) deprecation

---
 python/cugraph/cugraph/community/ktruss_subgraph.py      | 9 ++++-----
 .../cugraph/tests/community/test_k_truss_subgraph.py     | 4 ++--
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/python/cugraph/cugraph/community/ktruss_subgraph.py b/python/cugraph/cugraph/community/ktruss_subgraph.py
index 15a10007610..1799c50252f 100644
--- a/python/cugraph/cugraph/community/ktruss_subgraph.py
+++ b/python/cugraph/cugraph/community/ktruss_subgraph.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2019-2023, NVIDIA CORPORATION.
+# Copyright (c) 2019-2024, NVIDIA CORPORATION.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -95,11 +95,11 @@ def k_truss(
     G, isNx = ensure_cugraph_obj_for_nx(G)
 
     if isNx is True:
-        k_sub = ktruss_subgraph(G, k)
+        k_sub = ktruss_subgraph(G, k, use_weights=False)
         S = cugraph_to_nx(k_sub)
         return S
     else:
-        return ktruss_subgraph(G, k)
+        return ktruss_subgraph(G, k, use_weights=False)
 
 
 # FIXME: merge this function with k_truss
@@ -174,8 +174,7 @@ def ktruss_subgraph(
     --------
     >>> from cugraph.datasets import karate
     >>> G = karate.get_graph(download=True)
-    >>> k_subgraph = cugraph.ktruss_subgraph(G, 3)
-
+    >>> k_subgraph = cugraph.ktruss_subgraph(G, 3, use_weights=False)
     """
 
     _ensure_compatible_cuda_version()
diff --git a/python/cugraph/cugraph/tests/community/test_k_truss_subgraph.py b/python/cugraph/cugraph/tests/community/test_k_truss_subgraph.py
index c1f8f4c3546..063d7fc735f 100644
--- a/python/cugraph/cugraph/tests/community/test_k_truss_subgraph.py
+++ b/python/cugraph/cugraph/tests/community/test_k_truss_subgraph.py
@@ -1,4 +1,4 @@
-# Copyright (c) 2020-2023, NVIDIA CORPORATION.
+# Copyright (c) 2020-2024, NVIDIA CORPORATION.
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
@@ -98,7 +98,7 @@ def test_ktruss_subgraph_Graph(_, nx_ground_truth):
 
     k = 5
     G = polbooks.get_graph(download=True, create_using=cugraph.Graph(directed=False))
-    k_subgraph = cugraph.ktruss_subgraph(G, k)
+    k_subgraph = cugraph.ktruss_subgraph(G, k, use_weights=False)
 
     compare_k_truss(k_subgraph, k, nx_ground_truth)