Upload feature_extraction_moment.py

feature_extraction_moment.py

@@ -42,10 +42,10 @@ class MomentFeatureExtractor(FeatureExtractionMixin):
         return_tensors: Optional[Union[str, TensorType]] = TensorType.PYTORCH,
         torch_dtype: Optional[Union[str, torch.dtype]] = torch.float,
         padding: Union[bool, str] = False,  # 追加
-        max_length: int = None,   # 追加
+        max_length: Union[int, None] = None,   # 追加
     ) -> BatchFeature:
         if time_series is not None:
-            time_series_values, input_mask = self._convert_time_series(time_series, return_tensors, torch_dtype, max_length)
+            time_series_values, input_mask = self._convert_time_series(time_series, return_tensors, torch_dtype, padding, max_length)
         else:
             time_series_values = None
             input_mask = None