Upload feature_extraction_moment.py
Browse files
feature_extraction_moment.py
CHANGED
|
@@ -40,7 +40,7 @@ class MomentFeatureExtractor(FeatureExtractionMixin):
|
|
| 40 |
else:
|
| 41 |
time_series_values = None
|
| 42 |
|
| 43 |
-
return BatchFeature(data=time_series_values)
|
| 44 |
|
| 45 |
|
| 46 |
def _convert_time_series(self, time_series, return_tensors, torch_dtype):
|
|
|
|
| 40 |
else:
|
| 41 |
time_series_values = None
|
| 42 |
|
| 43 |
+
return BatchFeature(data={"time_series_values": time_series_values})
|
| 44 |
|
| 45 |
|
| 46 |
def _convert_time_series(self, time_series, return_tensors, torch_dtype):
|