Skip to content

Commit c7781fe

Browse files
elisimraghavanone
authored andcommitted
[Time-Series] fix past_observed_mask type (huggingface#22076)
added > 0.5 to `past_observed_mask`
1 parent 89c2557 commit c7781fe

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

tests/models/informer/test_modeling_informer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def prepare_informer_inputs_dict(self, config):
117117

118118
past_time_features = floats_tensor([self.batch_size, _past_length, config.num_time_features])
119119
past_values = floats_tensor([self.batch_size, _past_length])
120-
past_observed_mask = floats_tensor([self.batch_size, _past_length])
120+
past_observed_mask = floats_tensor([self.batch_size, _past_length]) > 0.5
121121

122122
# decoder inputs
123123
future_time_features = floats_tensor([self.batch_size, config.prediction_length, config.num_time_features])

tests/models/time_series_transformer/test_modeling_time_series_transformer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -114,7 +114,7 @@ def prepare_time_series_transformer_inputs_dict(self, config):
114114

115115
past_time_features = floats_tensor([self.batch_size, _past_length, config.num_time_features])
116116
past_values = floats_tensor([self.batch_size, _past_length])
117-
past_observed_mask = floats_tensor([self.batch_size, _past_length])
117+
past_observed_mask = floats_tensor([self.batch_size, _past_length]) > 0.5
118118

119119
# decoder inputs
120120
future_time_features = floats_tensor([self.batch_size, config.prediction_length, config.num_time_features])

0 commit comments

Comments
 (0)