remove debug code

rapidsai · Aug 7, 2024 · 29be85c · 29be85c
1 parent 41a4c39
commit 29be85c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 14 deletions.
diff --git a/python/cugraph-dgl/cugraph_dgl/graph.py b/python/cugraph-dgl/cugraph_dgl/graph.py
@@ -542,8 +542,7 @@ def _graph(
                 if prob_attr is None
                 else cupy.asarray(self.edata[prob_attr][edgelist_dict["eid"].cpu()])
             )
-            weights = cupy.array([1,1,2,0,0,0,2,1], dtype='float32')
-            print(weights)
+            weights = cupy.array([1, 1, 2, 0, 0, 0, 2, 1], dtype="float32")
 
             if self.is_multi_gpu:
                 rank = torch.distributed.get_rank()
@@ -618,9 +617,6 @@ def _get_n_emb(
             )
 
         try:
-            print(
-                u,
-            )
             return self.__ndata_storage[ntype, emb_name].fetch(
                 _cast_to_torch_tensor(u), "cuda"
             )

diff --git a/python/cugraph/cugraph/gnn/data_loading/dist_sampler.py b/python/cugraph/cugraph/gnn/data_loading/dist_sampler.py
@@ -638,7 +638,11 @@ def sample_from_nodes(
                 : len(current_seeds)
             ]
 
-            print(current_seeds, current_batches, flush=True,)
+            print(
+                current_seeds,
+                current_batches,
+                flush=True,
+            )
             minibatch_dict = self.sample_batches(
                 seeds=current_seeds,
                 batch_ids=current_batches,
@@ -768,14 +772,6 @@ def sample_batches(
             else:
                 label_offsets = None
 
-            print(
-                cupy.asarray(seeds),
-                cupy.asarray(batch_ids),
-                cupy.asarray(label_list),
-                cupy.asarray(label_to_output_comm_rank),
-                flush=True,
-            )
-
             sampling_results_dict = self.__func(
                 self._resource_handle,
                 self._graph,