remove span qa

allenai · Apr 5, 2024 · 28417a2 · 28417a2
1 parent af5d4df
commit 28417a2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git 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
@@ -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'
 ]