bypass attr

jkobject · Jul 17, 2024 · 5aaeb26 · 5aaeb26
1 parent 36c7b37
commit 5aaeb26
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/scdataloader/utils.py b/scdataloader/utils.py
@@ -138,11 +138,12 @@ def getBiomartTable(
     res.columns = attr + attributes
     if type(res) is not type(pd.DataFrame()):
         raise ValueError("should be a dataframe")
-    res = res[~(res["ensembl_gene_id"].isna() & res["hgnc_symbol"].isna())]
-    res.loc[res[res.hgnc_symbol.isna()].index, "hgnc_symbol"] = res[
-        res.hgnc_symbol.isna()
-    ]["ensembl_gene_id"]
-
+    res = res[~(res["ensembl_gene_id"].isna())]
+    if "hgnc_symbol" in res.columns:
+        res = res[res["hgnc_symbol"].isna()]
+        res.loc[res[res.hgnc_symbol.isna()].index, "hgnc_symbol"] = res[
+            res.hgnc_symbol.isna()
+        ]["ensembl_gene_id"]
     return res