From adc94c50535f60e5b1e1b19650aff455004c0973 Mon Sep 17 00:00:00 2001 From: Alexandria Barghi Date: Fri, 17 Nov 2023 11:18:39 -0800 Subject: [PATCH] remove unncessary argument from function --- python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py b/python/cugraph-pyg/cugraph_pyg/data/cugraph_store.py index b208e080de6..14dc5d84f90 100644 --- 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"])