Spaces:
Build error
Build error
File size: 1,276 Bytes
7e3e85d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
from model.base_model import SummModel
class SingleDocSummModel(SummModel):
def __init__(
self,
trained_domain: str = None,
max_input_length: int = None,
max_output_length: int = None,
):
super(SingleDocSummModel, self).__init__(
trained_domain=trained_domain,
max_input_length=max_input_length,
max_output_length=max_output_length,
)
@classmethod
def assert_summ_input_type(cls, corpus, query):
if not isinstance(corpus, list):
raise TypeError(
"Single-document summarization requires corpus of `List[str]`."
)
if not all([isinstance(ins, str) for ins in corpus]):
raise TypeError(
"Single-document summarization requires corpus of `List[str]`."
)
if query is not None:
if not isinstance(query, list):
raise TypeError(
"Query-based single-document summarization requires query of `List[str]`."
)
if not all([isinstance(q, str) for q in query]):
raise TypeError(
"Query-based single-document summarization requires query of `List[str]`."
)
|