File tree 4 files changed +15
-8
lines changed
4 files changed +15
-8
lines changed Original file line number Diff line number Diff line change @@ -16,7 +16,9 @@ class PreprocessingSuite:
16
16
param_names = ["input_data" ]
17
17
18
18
def setup (self , input_data : str ):
19
- self .adata = rsc .get .anndata_to_GPU (self ._data_dict [input_data ].copy (), copy = True )
19
+ self .adata = rsc .get .anndata_to_GPU (
20
+ self ._data_dict [input_data ].copy (), copy = True
21
+ )
20
22
21
23
def time_calculate_qc_metrics (self , * _ ):
22
24
self .adata .var ["mt" ] = self .adata .var_names .str .startswith ("MT-" )
Original file line number Diff line number Diff line change 38
38
39
39
sc .settings .datasetdir = pathlib .Path (__file__ ).parent .resolve () / "data"
40
40
41
+
41
42
@dataclass
42
43
class Dataset :
43
44
path : Path
44
45
get : Callable [[], anndata .AnnData ]
45
46
46
- path = "/p/project/training2406/team_scverse/gold2/rapids_singlecell/benchmarks/data/pbmc3k_raw.h5ad"
47
+
48
+ path = "/p/project/training2406/team_scverse/gold2/rapids_singlecell/benchmarks/data/pbmc3k_raw.h5ad"
47
49
48
50
49
51
class ToGPUSuite :
Original file line number Diff line number Diff line change 11
11
12
12
import rapids_singlecell as rsc
13
13
14
- import pathlib
15
14
16
15
class ToolsSuite :
17
16
_data_dict = dict (
18
- visium_sge = ad .read_h5ad ("/p/project/training2406/team_scverse/gold2/rapids_singlecell/benchmarks/data/paul15.h5ad" ),
17
+ visium_sge = ad .read_h5ad (
18
+ "/p/project/training2406/team_scverse/gold2/rapids_singlecell/benchmarks/data/paul15.h5ad"
19
+ ),
19
20
)
20
21
params = _data_dict .keys ()
21
22
param_names = ["input_data" ]
22
23
23
24
def setup (self , input_data ):
24
- self .adata = rsc .get .anndata_to_GPU (self ._data_dict [input_data ].copy (), copy = True )
25
+ self .adata = rsc .get .anndata_to_GPU (
26
+ self ._data_dict [input_data ].copy (), copy = True
27
+ )
25
28
26
29
def time_ligrec (self , * _ ):
27
30
gene_ids = self .adata .var .index
Original file line number Diff line number Diff line change 9
9
10
10
import rapids_singlecell as rsc
11
11
12
- import pathlib
13
-
14
12
15
13
class ToolsSuite :
16
14
_data_dict = dict (
@@ -20,7 +18,9 @@ class ToolsSuite:
20
18
param_names = ["input_data" ]
21
19
22
20
def setup (self , input_data ):
23
- self .adata = rsc .get .anndata_to_GPU (self ._data_dict [input_data ].copy (), copy = True )
21
+ self .adata = rsc .get .anndata_to_GPU (
22
+ self ._data_dict [input_data ].copy (), copy = True
23
+ )
24
24
assert "X_pca" in self .adata .obsm
25
25
26
26
def time_umap (self , * _ ):
You can’t perform that action at this time.
0 commit comments