Error handling

scverse · Feb 16, 2019 · a8c15ce · a8c15ce
1 parent 9131ac0
commit a8c15ce
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/scanpy/queries/_queries.py b/scanpy/queries/_queries.py
@@ -150,10 +150,14 @@ def mitochondrial_genes(org, attrname="external_gene_name", host="www.ensembl.or
 @singledispatch
 def enrich(container, org: Optional[str] = None) -> pd.DataFrame:
     """Get enrichment for DE results."""
-    from gprofiler import gprofiler
-    return gprofiler(container, org)
+    try:
+        from gprofiler import gprofiler
+    except ImportError:
+        raise ImportError("You need to install the `gprofiler` module.")
+    return pd.DataFrame(gprofiler(container, org))
 
 
 @enrich.register
-def _enrich_anndata(adata: AnnData, group, key: str = "rank_genes_groups", org: Optional[str] = None) -> pd.DataFrame:
+def _enrich_anndata(adata: AnnData, group, key: str = "rank_genes_groups",
+                    org: Optional[str] = None) -> pd.DataFrame:
     return enrich(adata.uns[key]["names"][group], org=org)