diff --git a/src/gluonts/torch/model/mq_cnn/estimator.py b/src/gluonts/torch/model/mq_cnn/estimator.py index f678619dca..b0d779a151 100644 --- a/src/gluonts/torch/model/mq_cnn/estimator.py +++ b/src/gluonts/torch/model/mq_cnn/estimator.py @@ -619,13 +619,10 @@ def _create_instance_splitter(self, mode: str) -> Chain: else [] ) - decoder_series_fields = ( - [ - FieldName.FEAT_DYNAMIC, - FieldName.FEAT_DYNAMIC_CAT, - ] - + ([FieldName.OBSERVED_VALUES] if mode != "test" else []) - ) + decoder_series_fields = [ + FieldName.FEAT_DYNAMIC, + FieldName.FEAT_DYNAMIC_CAT, + ] + ([FieldName.OBSERVED_VALUES] if mode != "test" else []) decoder_disabled_fields = ( [FieldName.FEAT_DYNAMIC, FieldName.FEAT_DYNAMIC_CAT]