davidjwen commited on
Commit
deb9c9d
1 Parent(s): e3330a6

Upload in_silico_perturber.py

Browse files
Files changed (1) hide show
  1. geneformer/in_silico_perturber.py +1 -1
geneformer/in_silico_perturber.py CHANGED
@@ -636,7 +636,7 @@ class InSilicoPerturber:
636
  if len(self.genes_to_perturb) > 1:
637
  tokens_to_perturb = tuple(self.tokens_to_perturb)
638
  else:
639
- tokens_to_perturb = self.tokens_to_perturb
640
 
641
  # fill in the gene cosine similarities
642
  try:
 
636
  if len(self.genes_to_perturb) > 1:
637
  tokens_to_perturb = tuple(self.tokens_to_perturb)
638
  else:
639
+ tokens_to_perturb = self.tokens_to_perturb[0]
640
 
641
  # fill in the gene cosine similarities
642
  try: