remove unncessary argument from function

rapidsai · Nov 17, 2023 · adc94c5 · adc94c5
1 parent 4c9b73b
commit adc94c5
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py b/python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py
@@ -368,9 +368,7 @@ def __infer_offsets(
             }
         )
 
-    def __dask_array_from_numpy(
-        self, array: np.ndarray, client: distributed.client.Client, npartitions: int
-    ):
+    def __dask_array_from_numpy(self, array: np.ndarray, npartitions: int):
         return dar.from_array(
             array,
             meta=np.array([], dtype=array.dtype),
@@ -481,13 +479,13 @@ def __construct_graph(
             nworkers = len(client.scheduler_info()["workers"])
             npartitions = nworkers * 4
 
-            src_dar = self.__dask_array_from_numpy(na_src, client, npartitions)
+            src_dar = self.__dask_array_from_numpy(na_src, npartitions)
             del na_src
 
-            dst_dar = self.__dask_array_from_numpy(na_dst, client, npartitions)
+            dst_dar = self.__dask_array_from_numpy(na_dst, npartitions)
             del na_dst
 
-            etp_dar = self.__dask_array_from_numpy(na_etp, client, npartitions)
+            etp_dar = self.__dask_array_from_numpy(na_etp, npartitions)
             del na_etp
 
             df = dd.from_dask_array(etp_dar, columns=["etp"])