ctheodoris
/

Geneformer

Inference Endpoints

Model card Files Files and versions Community

davidjwen commited on Jan 21

Commit

deb9c9d

•

1 Parent(s): e3330a6

Upload in_silico_perturber.py

Files changed (1) hide show

geneformer/in_silico_perturber.py +1 -1

geneformer/in_silico_perturber.py CHANGED Viewed

@@ -636,7 +636,7 @@ class InSilicoPerturber:
                             if len(self.genes_to_perturb) > 1:
                                 tokens_to_perturb = tuple(self.tokens_to_perturb)
                             else:
-                                tokens_to_perturb = self.tokens_to_perturb
                             # fill in the gene cosine similarities
                             try:

                             if len(self.genes_to_perturb) > 1:
                                 tokens_to_perturb = tuple(self.tokens_to_perturb)
                             else:
+                                tokens_to_perturb = self.tokens_to_perturb[0]
                             # fill in the gene cosine similarities
                             try: