Upload processing_time_rcd.py
Browse files- processing_time_rcd.py +5 -3
processing_time_rcd.py
CHANGED
|
@@ -57,12 +57,14 @@ class TimeRCDProcessor(ProcessorMixin):
|
|
| 57 |
self.normalize = normalize
|
| 58 |
self.pad_to_multiple = pad_to_multiple
|
| 59 |
|
| 60 |
-
# Required attributes for ProcessorMixin
|
| 61 |
-
self.model_input_names = ["time_series", "attention_mask"]
|
| 62 |
-
|
| 63 |
# Call parent init after setting attributes
|
| 64 |
super().__init__(**kwargs)
|
| 65 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 66 |
@property
|
| 67 |
def attributes(self):
|
| 68 |
"""Return list of attribute names for serialization."""
|
|
|
|
| 57 |
self.normalize = normalize
|
| 58 |
self.pad_to_multiple = pad_to_multiple
|
| 59 |
|
|
|
|
|
|
|
|
|
|
| 60 |
# Call parent init after setting attributes
|
| 61 |
super().__init__(**kwargs)
|
| 62 |
|
| 63 |
+
@property
|
| 64 |
+
def model_input_names(self):
|
| 65 |
+
"""Return list of model input names."""
|
| 66 |
+
return ["time_series", "attention_mask"]
|
| 67 |
+
|
| 68 |
@property
|
| 69 |
def attributes(self):
|
| 70 |
"""Return list of attribute names for serialization."""
|