Christina Theodoris
commited on
Commit
•
8c2fae7
1
Parent(s):
67b7f01
Fix bug in clearing cache
Browse files
geneformer/in_silico_perturber.py
CHANGED
|
@@ -763,8 +763,9 @@ class InSilicoPerturber:
|
|
| 763 |
del cell_data
|
| 764 |
del data_list
|
| 765 |
elif self.anchor_token is None:
|
| 766 |
-
|
| 767 |
-
|
|
|
|
| 768 |
else:
|
| 769 |
del combo_cos_sim
|
| 770 |
del combo_cos_sims_data
|
|
|
|
| 763 |
del cell_data
|
| 764 |
del data_list
|
| 765 |
elif self.anchor_token is None:
|
| 766 |
+
if self.emb_mode == "cell_and_gene":
|
| 767 |
+
del affected_gene
|
| 768 |
+
del cos_sim_value
|
| 769 |
else:
|
| 770 |
del combo_cos_sim
|
| 771 |
del combo_cos_sims_data
|