From 83a6f99b0217f0086f218e524f86dab0a6840286 Mon Sep 17 00:00:00 2001 From: "Wesley M. Gifford" Date: Fri, 22 Mar 2024 13:46:33 -0400 Subject: [PATCH] pass context length Signed-off-by: Wesley M. Gifford --- tsfm_public/toolkit/time_series_preprocessor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tsfm_public/toolkit/time_series_preprocessor.py b/tsfm_public/toolkit/time_series_preprocessor.py index eb598047..eb8137b0 100644 --- a/tsfm_public/toolkit/time_series_preprocessor.py +++ b/tsfm_public/toolkit/time_series_preprocessor.py @@ -623,7 +623,7 @@ def get_datasets( # get split_params # split_params = get_split_params(config, self.context_length, len(data)) - split_params, split_function = get_split_params(split_config) + split_params, split_function = get_split_params(split_config, context_length=self.context_length) # specify columns column_specifiers = {