pass context length

Signed-off-by: Wesley M. Gifford <[email protected]>
ibm-granite · Mar 22, 2024 · 83a6f99 · 83a6f99
1 parent ccba0ea
commit 83a6f99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 = {