diff --git a/python/tasks/jigsaw_toxic.py b/python/tasks/jigsaw_toxic.py
index f1d0749..7b5310c 100644
--- a/python/tasks/jigsaw_toxic.py
+++ b/python/tasks/jigsaw_toxic.py
@@ -222,10 +222,8 @@ def kernel_2():
     return dict(
         model=model,
         xtrain_pad=xtrain_pad,
-        ytrain=ytrain,
         strategy=strategy,
         xvalid_pad=xvalid_pad,
-        yvalid=yvalid,
         xtrain_seq=xtrain_seq,
         token=token,
         max_len=max_len,