diff --git a/tsfm_public/toolkit/dataset.py b/tsfm_public/toolkit/dataset.py
index 85098a88..27f782b3 100644
--- a/tsfm_public/toolkit/dataset.py
+++ b/tsfm_public/toolkit/dataset.py
@@ -346,7 +346,6 @@ def __init__(
         num_workers: int = 1,
         frequency_token: Optional[int] = None,
         autoregressive_modeling: bool = True,
-        training: bool = True,
     ):
         # output_columns_tmp = input_columns if output_columns == [] else output_columns