Skip to content

Commit 6ebac49

Browse files
fix index error and add grn folder
1 parent 19a50e6 commit 6ebac49

1 file changed

Lines changed: 4 additions & 3 deletions

File tree

src/netmap/grn/inferrence.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import torch
55
from tqdm import tqdm
66

7-
import pandas as pd
7+
import pandas as pdimport pyarrow.dataset as ds
88

99
from netmap.utils.data_utils import attribution_to_anndata
1010
import itertools
@@ -287,16 +287,17 @@ def inferrence(models, data_train_full_tensor, gene_names, xai_method='GradientS
287287

288288
# Configuration
289289
output_dir = op.dirname(backing_file)
290-
op.join(output_dir, 'grn')
290+
output_dir = op.join(output_dir, 'grn')
291291
os.makedirs(output_dir, exist_ok=True)
292292

293-
name_list = list(gene_names)
293+
name_list = []
294294
name = 'attr'
295295

296296
for i in range(cols):
297297
## Create name vector
298298
name_list = name_list + list(gene_names)
299299
target_names = target_names+[gene_names[i]] *len(gene_names)
300+
300301
column_names = [f'{s}_{t}' for s,t in zip(name_list, target_names)]
301302

302303
schema = pa.schema([(name, pa.float32()) for name in column_names])

0 commit comments

Comments
 (0)