File tree Expand file tree Collapse file tree 4 files changed +15
-8
lines changed Expand file tree Collapse file tree 4 files changed +15
-8
lines changed Original file line number Diff line number Diff line change @@ -16,7 +16,9 @@ class PreprocessingSuite:
1616 param_names = ["input_data" ]
1717
1818 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+ )
2022
2123 def time_calculate_qc_metrics (self , * _ ):
2224 self .adata .var ["mt" ] = self .adata .var_names .str .startswith ("MT-" )
Original file line number Diff line number Diff line change 3838
3939sc .settings .datasetdir = pathlib .Path (__file__ ).parent .resolve () / "data"
4040
41+
4142@dataclass
4243class Dataset :
4344 path : Path
4445 get : Callable [[], anndata .AnnData ]
4546
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"
4749
4850
4951class ToGPUSuite :
Original file line number Diff line number Diff line change 1111
1212import rapids_singlecell as rsc
1313
14- import pathlib
1514
1615class ToolsSuite :
1716 _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+ ),
1920 )
2021 params = _data_dict .keys ()
2122 param_names = ["input_data" ]
2223
2324 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+ )
2528
2629 def time_ligrec (self , * _ ):
2730 gene_ids = self .adata .var .index
Original file line number Diff line number Diff line change 99
1010import rapids_singlecell as rsc
1111
12- import pathlib
13-
1412
1513class ToolsSuite :
1614 _data_dict = dict (
@@ -20,7 +18,9 @@ class ToolsSuite:
2018 param_names = ["input_data" ]
2119
2220 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+ )
2424 assert "X_pca" in self .adata .obsm
2525
2626 def time_umap (self , * _ ):
You can’t perform that action at this time.
0 commit comments