From f02b4dac09b76cce39e8cba6c785186de289fcd8 Mon Sep 17 00:00:00 2001 From: Parmeet Singh Bhatia Date: Tue, 17 May 2022 09:43:56 -0400 Subject: [PATCH] fix docs build --- .../tutorials/sst2_classification_non_distributed.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/examples/tutorials/sst2_classification_non_distributed.py b/examples/tutorials/sst2_classification_non_distributed.py index 983735ef35..e3d1f77133 100644 --- a/examples/tutorials/sst2_classification_non_distributed.py +++ b/examples/tutorials/sst2_classification_non_distributed.py @@ -116,14 +116,14 @@ def apply_transform(x): # # :: # -# def batch_transform(x): -# return {"token_ids": text_transform(x["text"]), "target": x["label"]} +# def batch_transform(x): +# return {"token_ids": text_transform(x["text"]), "target": x["label"]} # # -# train_datapipe = train_datapipe.batch(batch_size).rows2columnar(["text", "label"]) -# train_datapipe = train_datapipe.map(lambda x: batch_transform) -# dev_datapipe = dev_datapipe.batch(batch_size).rows2columnar(["text", "label"]) -# dev_datapipe = dev_datapipe.map(lambda x: batch_transform) +# train_datapipe = train_datapipe.batch(batch_size).rows2columnar(["text", "label"]) +# train_datapipe = train_datapipe.map(lambda x: batch_transform) +# dev_datapipe = dev_datapipe.batch(batch_size).rows2columnar(["text", "label"]) +# dev_datapipe = dev_datapipe.map(lambda x: batch_transform) # ######################################################################