From c63c7e23df8a806ddcdc105f3085d1c2b300949d Mon Sep 17 00:00:00 2001 From: elisim Date: Fri, 10 Mar 2023 14:51:46 +0700 Subject: [PATCH] added > 0.5 to `past_observed_mask` --- tests/models/informer/test_modeling_informer.py | 2 +- .../test_modeling_time_series_transformer.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/models/informer/test_modeling_informer.py b/tests/models/informer/test_modeling_informer.py index 91e6fb74f6f151..271f997bee8b63 100644 --- a/tests/models/informer/test_modeling_informer.py +++ b/tests/models/informer/test_modeling_informer.py @@ -117,7 +117,7 @@ def prepare_informer_inputs_dict(self, config): past_time_features = floats_tensor([self.batch_size, _past_length, config.num_time_features]) past_values = floats_tensor([self.batch_size, _past_length]) - past_observed_mask = floats_tensor([self.batch_size, _past_length]) + past_observed_mask = floats_tensor([self.batch_size, _past_length]) > 0.5 # decoder inputs future_time_features = floats_tensor([self.batch_size, config.prediction_length, config.num_time_features]) diff --git a/tests/models/time_series_transformer/test_modeling_time_series_transformer.py b/tests/models/time_series_transformer/test_modeling_time_series_transformer.py index 7f14f29de04090..65834dac42f78c 100644 --- a/tests/models/time_series_transformer/test_modeling_time_series_transformer.py +++ b/tests/models/time_series_transformer/test_modeling_time_series_transformer.py @@ -114,7 +114,7 @@ def prepare_time_series_transformer_inputs_dict(self, config): past_time_features = floats_tensor([self.batch_size, _past_length, config.num_time_features]) past_values = floats_tensor([self.batch_size, _past_length]) - past_observed_mask = floats_tensor([self.batch_size, _past_length]) + past_observed_mask = floats_tensor([self.batch_size, _past_length]) > 0.5 # decoder inputs future_time_features = floats_tensor([self.batch_size, config.prediction_length, config.num_time_features])