black

awslabs · May 27, 2024 · 2695b74 · 2695b74
1 parent ddbd57f
commit 2695b74
Showing 1 changed file with 4 additions and 7 deletions.
diff --git 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]