Add Summarization to Pipelines (#3128)

* passing

* Undo stupid chg

* docs

* undo rename

* delete-cruft

* only import if you have torch

* Dont rely on dict ordering

* Fix dict ordering upstream

* docstring link

* docstring link

* remove trailing comma for 3.5 compat

* new name

* delegate kwarging

* Update kwargs
This commit is contained in:
Sam Shleifer
2020-03-17 18:04:21 -04:00
committed by GitHub
parent 2b60a26b46
commit 38a555a83c
4 changed files with 129 additions and 1 deletions

View File

@@ -247,6 +247,16 @@ class MonoColumnInputTestCase(unittest.TestCase):
expected_check_keys=["sequence"],
)
@require_torch
def test_summarization(self):
valid_inputs = ["A string like this", ["list of strings entry 1", "list of strings v2"]]
invalid_inputs = [4, "<mask>"]
mandatory_keys = ["summary_text"]
nlp = pipeline(task="summarization")
self._test_mono_column_pipeline(
nlp, valid_inputs, invalid_inputs, mandatory_keys,
)
class MultiColumnInputTestCase(unittest.TestCase):
def _test_multicolumn_pipeline(self, nlp, valid_inputs: list, invalid_inputs: list, output_keys: Iterable[str]):