Merge pull request #107 from SBRG/avsastry-patch-1

Update core.py
SBRG · Feb 28, 2022 · ff46bcc · ff46bcc
2 parents 25c5814 + 9f30b6a
commit ff46bcc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pymodulon/core.py b/src/pymodulon/core.py
@@ -1129,7 +1129,7 @@ def imodulons_with(self, gene):
         """
 
         # Check that gene exists
-        if gene not in self.X.index:
+        if gene not in self.gene_names:
             gene = self.name2num(gene)
 
         return self.M.columns[self.M_binarized.loc[gene] == 1].to_list()