gluonts.torch.distributions.BinnedUniformsOutput raises AttributeError

[Drewniacki]

Description

Using BinnedUniformsOutput with DeepAREstimator raises AttributeError: 'list' object has no attribute 'shape'.

To Reproduce

model_params = {
    'prediction_length': pred_length,
    'freq': period.freqstr,
    'distr_output' : BinnedUniformsOutput(
                        bins_lower_bound = 0.0, 
                        bins_upper_bound = 1.0, 
                        num_bins=2),
}
model = DeepAREstimator(**model_params).train(train_ds)

Error message or code output

INFO: GPU available: False, used: False
INFO:lightning.pytorch.utilities.rank_zero:GPU available: False, used: False
INFO: TPU available: False, using: 0 TPU cores
INFO:lightning.pytorch.utilities.rank_zero:TPU available: False, using: 0 TPU cores
INFO: IPU available: False, using: 0 IPUs
INFO:lightning.pytorch.utilities.rank_zero:IPU available: False, using: 0 IPUs
INFO: HPU available: False, using: 0 HPUs
INFO:lightning.pytorch.utilities.rank_zero:HPU available: False, using: 0 HPUs
/usr/local/lib/python3.10/dist-packages/lightning/pytorch/trainer/configuration_validator.py:74: You defined a `validation_step` but have no `val_dataloader`. Skipping val loop.
---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
[<ipython-input-173-9cdb9f00c907>](https://localhost:8080/#) in <cell line: 1>()
----> 1 model = DeepAREstimator(**model_params).train(train_ds)

21 frames
[/usr/local/lib/python3.10/dist-packages/gluonts/torch/distributions/binned_uniforms.py](https://localhost:8080/#) in __init__(self, bins_lower_bound, bins_upper_bound, logits, numb_bins, validate_args)
     53         )
     54         assert (
---> 55             logits.shape[-1] == numb_bins
     56         ), "The distribution requires one logit per bin."
     57 

AttributeError: 'list' object has no attribute 'shape'

Environment

• Google Colab
• Python version: python 3.10
• GluonTS version: 0.14.3
• MXNet version: 1.9.1