ctheodoris
/

Geneformer

Inference Endpoints

Model card Files Files and versions Community

Upload in_silico_perturber.py

#292

by davidjwen - opened Jan 21

base: refs/heads/main

←

from: refs/pr/292

Discussion Files changed

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: