transformer/multivariate.ipynb GluonTSDataError

[kevindarby]

File ~/code/cqg/bts/spark/ml/.venv/lib/python3.9/site-packages/gluonts/torch/model/estimator.py:205, in PyTorchLightningEstimator.train_model(self, training_data, validation_data, from_predictor, shuffle_buffer_length, cache_data, ckpt_path, **kwargs)
...
93 """
94 if not condition:
---> 95 raise exception_class(message.format(*args, **kwargs))

GluonTSDataError: Input for field "target" does not have the requireddimension (field: target, ndim observed: 2, expected ndim: 1)

[kevindarby]

gluonts==0.13.1

[kevindarby]

happens on the train call

predictor = estimator.train(
training_data=dataset_train,
num_workers=8,
shuffle_buffer_length=1024
)

[kashif]

hmm i see let me confirm... i might need to update it for the new gluonts api

[kevindarby]

Thanks StudentTOutput changed too

[kashif]

indeed