fix from huggingface#22076

elisim · Mar 10, 2023 · 980c266 · 980c266
1 parent 3a87689
commit 980c266
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/autoformer/test_modeling_autoformer.py b/tests/models/autoformer/test_modeling_autoformer.py
@@ -117,7 +117,7 @@ def prepare_autoformer_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])