fix doc string example in TranslationData (#221)

* fix doc string example in TranslationData * Fix docs build Co-authored-by: Jirka Borovec <[email protected]> Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> Co-authored-by: Ethan Harris <[email protected]>
Lightning-Universe · May 4, 2021 · ec404e8 · ec404e8
1 parent 54e569b
commit ec404e8
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/flash/text/seq2seq/translation/data.py b/flash/text/seq2seq/translation/data.py
@@ -64,10 +64,14 @@ def from_files(
 
         Examples::
 
-            train_df = pd.read_csv("train_data.csv")
-            tab_data = TabularData.from_df(train_df, target="fraud",
-                                           num_cols=["account_value"],
-                                           cat_cols=["account_type"])
+            datamodule = TranslationData.from_files(
+                train_file="data/wmt_en_ro/train.csv",
+                val_file="data/wmt_en_ro/valid.csv",
+                test_file="data/wmt_en_ro/test.csv",
+                input="input",
+                target="target",
+                batch_size=1,
+            )
 
         """
         return super().from_files(