Docstrings for seq2seq data

CHANGELOG.md

@@ -64,8 +64,12 @@ The format is based on [Keep a Changelog](http://keepachangelog.com/en/1.0.0/).
 
 - Fixed a bug when loading prediction data for graph classification without targets ([#1121](https://github.com/PyTorchLightning/lightning-flash/pull/1121))
 
+- Fixed a bug where loading Seq2Seq data for prediction would not work if the target field was not present ([#1128](https://github.com/PyTorchLightning/lightning-flash/pull/1128))
+
 ### Removed
 
+- Removed the `Seq2SeqData` base class (use `TranslationData` or `SummarizationData` directly) ([#1128](https://github.com/PyTorchLightning/lightning-flash/pull/1128))
+
 ## [0.6.0] - 2021-13-12
 
 ### Added

docs/source/api/text.rst

@@ -77,7 +77,6 @@ _______________
     :template: classtemplate.rst
 
     ~seq2seq.core.model.Seq2SeqTask
-    ~seq2seq.core.data.Seq2SeqData
 
     seq2seq.core.input.Seq2SeqInputBase
     seq2seq.core.input.Seq2SeqCSVInput

flash/text/__init__.py

@@ -2,7 +2,6 @@
 from flash.text.embedding import TextEmbedder  # noqa: F401
 from flash.text.question_answering import QuestionAnsweringData, QuestionAnsweringTask  # noqa: F401
 from flash.text.seq2seq import (  # noqa: F401
-    Seq2SeqData,
     Seq2SeqTask,
     SummarizationData,
     SummarizationTask,

flash/text/seq2seq/__init__.py

@@ -1,3 +1,3 @@
-from flash.text.seq2seq.core import Seq2SeqData, Seq2SeqTask  # noqa: F401
+from flash.text.seq2seq.core import Seq2SeqTask  # noqa: F401
 from flash.text.seq2seq.summarization import SummarizationData, SummarizationTask  # noqa: F401
 from flash.text.seq2seq.translation import TranslationData, TranslationTask  # noqa: F401

flash/text/seq2seq/core/__init__.py

@@ -1,2 +1 @@
-from flash.text.seq2seq.core.data import Seq2SeqData  # noqa: F401
 from flash.text.seq2seq.core.model import Seq2SeqTask  # noqa: F401

flash/text/seq2seq/core/input.py

@@ -47,7 +47,7 @@ def load_data(
         if input_key != DataKeys.INPUT:
             hf_dataset = hf_dataset.rename_column(input_key, DataKeys.INPUT)
 
-        if target_key != DataKeys.TARGET:
+        if target_key in hf_dataset.column_names and target_key != DataKeys.TARGET:
             hf_dataset = hf_dataset.rename_column(target_key, DataKeys.TARGET)
 
         if flash._IS_TESTING: