diff --git a/pori_python/ipr/annotate.py b/pori_python/ipr/annotate.py index 9a85d3e..a7f20d3 100644 --- a/pori_python/ipr/annotate.py +++ b/pori_python/ipr/annotate.py @@ -11,7 +11,7 @@ from pori_python.graphkb.types import Variant from pori_python.graphkb.util import FeatureNotFoundError from pandas import isnull -from progressbar import progressbar +from tqdm import tqdm from typing import Dict, List, Sequence from .constants import TMB_HIGH_CATEGORY @@ -102,7 +102,7 @@ def annotate_expression_variants( problem_genes = set() logger.info(f"Starting annotation of {len(variants)} expression category_variants") - iterfunc = progressbar if show_progress else iter + iterfunc = tqdm if show_progress else iter for row in iterfunc(variants): gene = row['gene'] variant = row['variant'] @@ -155,7 +155,7 @@ def annotate_copy_variants( problem_genes = set() logger.info(f"Starting annotation of {len(variants)} copy category_variants") - iterfunc = progressbar if show_progress else iter + iterfunc = tqdm if show_progress else iter for row in iterfunc(variants): gene = row['gene'] variant = row['variant'] @@ -213,7 +213,7 @@ def annotate_positional_variants( alterations = [] problem_genes = set() - iterfunc = progressbar if show_progress else iter + iterfunc = tqdm if show_progress else iter for row in iterfunc(variants): if not row.get('gene') and (not row.get('gene1') or not row.get('gene2')): # https://www.bcgsc.ca/jira/browse/GERO-56?focusedCommentId=1234791&page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel#comment-1234791 diff --git a/setup.cfg b/setup.cfg index c69b087..b26b858 100644 --- a/setup.cfg +++ b/setup.cfg @@ -32,10 +32,10 @@ dependency_links = [] include_package_data = True install_requires = biopython - progressbar2 - pandas>=1.1.0 jsonschema + pandas>=1.1.0 requests + tqdm typing_extensions>=3.7.4.2,<5 [options.extras_require]