From 28417a29d0cc7dbc5e369bffee6cf9cb02af77d3 Mon Sep 17 00:00:00 2001 From: kyleclo Date: Fri, 5 Apr 2024 12:25:07 -0700 Subject: [PATCH] remove span qa --- papermage/predictors/__init__.py | 5 +++-- pyproject.toml | 6 ++++-- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/papermage/predictors/__init__.py b/papermage/predictors/__init__.py index ae62f67..038ef40 100644 --- a/papermage/predictors/__init__.py +++ b/papermage/predictors/__init__.py @@ -3,7 +3,8 @@ from papermage.predictors.block_predictors import LPEffDetPubLayNetBlockPredictor from papermage.predictors.formula_predictors import LPEffDetFormulaPredictor from papermage.predictors.sentence_predictors import PysbdSentencePredictor -from papermage.predictors.span_qa_predictors import APISpanQAPredictor + +# from papermage.predictors.span_qa_predictors import APISpanQAPredictor from papermage.predictors.token_predictors import HFWhitspaceTokenPredictor from papermage.predictors.vila_predictors import IVILATokenClassificationPredictor from papermage.predictors.word_predictors import SVMWordPredictor @@ -16,6 +17,6 @@ "PysbdSentencePredictor", "LPEffDetPubLayNetBlockPredictor", "LPEffDetFormulaPredictor", - "APISpanQAPredictor", + # "APISpanQAPredictor", "BasePredictor", ] diff --git a/pyproject.toml b/pyproject.toml index fc20630..14588dd 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [project] name = 'papermage' -version = '0.19.0' +version = '0.20.0' description = 'Papermage. Casting magic over scientific PDFs.' license = {text = 'Apache-2.0'} readme = 'README.md' @@ -131,9 +131,11 @@ predictors = [ 'wandb>=0.15.7', 'seqeval==1.2.2', 'effdet==0.3.0', - 'decontext==0.1.6', 'vila==0.5.0' ] +decontext = [ + 'decontext==0.1.6' +] production = [ 'optimum[onnxruntime]==1.10.0' ]