From 7f2da4fc8952262f2067bb7bfd4e15747c4e3e55 Mon Sep 17 00:00:00 2001
From: Laura Luebbert <laura.lbt60@gmail.com>
Date: Fri, 11 Nov 2022 12:26:09 -0800
Subject: [PATCH] Speed increase gget enrichr and archs4 with Ensembl IDs

---
 gget/__init__.py     | 2 +-
 gget/gget_archs4.py  | 4 ++--
 gget/gget_enrichr.py | 5 +++--
 3 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/gget/__init__.py b/gget/__init__.py
index 2ba1cfef..7f707499 100644
--- a/gget/__init__.py
+++ b/gget/__init__.py
@@ -20,6 +20,6 @@
 # Mute numexpr threads info
 logging.getLogger("numexpr").setLevel(logging.WARNING)
 
-__version__ = "0.3.12"
+__version__ = "0.3.13"
 __author__ = "Laura Luebbert"
 __email__ = "lauraluebbert@caltech.edu"
diff --git a/gget/gget_archs4.py b/gget/gget_archs4.py
index 422eb079..b4cf2a8a 100644
--- a/gget/gget_archs4.py
+++ b/gget/gget_archs4.py
@@ -71,8 +71,8 @@ def archs4(
     if ensembl:
         # Remove version number if passed
         gene = gene.split(".")[0]
-        
-        info_df = info(gene, verbose=False)
+
+        info_df = info(gene, verbose=False, ensembl_only=True)
 
         # Check if Ensembl ID was found
         if isinstance(info_df, type(None)):
diff --git a/gget/gget_enrichr.py b/gget/gget_enrichr.py
index 3cb724e5..7342866b 100644
--- a/gget/gget_enrichr.py
+++ b/gget/gget_enrichr.py
@@ -30,7 +30,7 @@ def enrichr(
     database,
     ensembl=False,
     plot=False,
-    figsize=(10,10),
+    figsize=(10, 10),
     ax=None,
     json=False,
     save=False,
@@ -110,11 +110,12 @@ def enrichr(
         logging.info("Getting gene symbols from Ensembl IDs.")
 
         genes_v2 = []
+
         for gene_id in genes:
             # Remove version number if passed
             gene_id = gene_id.split(".")[0]
 
-            info_df = info(gene_id, verbose=False)
+            info_df = info(gene_id, ensembl_only=True, verbose=False)
 
             # Check if Ensembl ID was found
             if isinstance(info_df, type(None)):