Upload feature_extraction_moment.py

feature_extraction_moment.py

@@ -188,7 +188,7 @@ class MomentFeatureExtractor(FeatureExtractionMixin):
         device = time_series_values[0].device
 
         padded_values = torch.zeros((batch_size, n_channels, target_len), dtype=time_series_values[0].dtype, device=device)
-        input_mask = torch.zeros((batch_size, target_len), dtype=torch.bool, device=device)
+        input_mask = torch.zeros((batch_size, target_len), dtype=time_series_values[0].dtype, device=device)
         
         for i in range(batch_size):
             seq = time_series_values[i]