add annotations

Co-authored-by: Tingyu Wang <[email protected]>
rapidsai · Sep 25, 2023 · 6990c23 · 6990c23
1 parent a124964
commit 6990c23
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/cugraph/cugraph/sampling/uniform_neighbor_sample.py b/python/cugraph/cugraph/sampling/uniform_neighbor_sample.py
@@ -71,9 +71,9 @@ def uniform_neighbor_sample(
     prior_sources_behavior: str = None,
     deduplicate_sources: bool = False,
     renumber: bool = False,
-    use_legacy_names=True,  # deprecated
-    compress_per_hop=False,
-    compression="COO",
+    use_legacy_names: bool = True,  # deprecated
+    compress_per_hop: bool = False,
+    compression: str = "COO",
 ) -> Union[cudf.DataFrame, Tuple[cudf.DataFrame, cudf.DataFrame]]:
     """
     Does neighborhood sampling, which samples nodes from a graph based on the