def test_generate(recipe) -> None: start = pd.Timestamp("2014-01-01", freq="D") result = take_as_list(iterator=generate(length=10, recipe=BASE_RECIPE, start=start), num=10) assert len(result) == 10
def generate(self) -> TrainDatasets: metadata = self.metadata data_it = generate( length=self.max_train_length + self.prediction_length, recipe=self.recipe, start=self.data_start, ) full_length_data = take_as_list(data_it, self.num_timeseries) test_data = [ RecipeDataset.trim_ts_item_front( x, self.trim_length_fun(x, train_length=self.max_train_length)) for x in full_length_data ] train_data = [ RecipeDataset.trim_ts_item_end(x, self.prediction_length) for x in test_data ] return TrainDatasets( metadata=metadata, train=ListDataset(train_data, metadata.time_granularity), test=ListDataset(test_data, metadata.time_granularity), )