Import error in pts module: gluonts/nursery/robust-mts-attack/pts/modules/distribution_output.py

[ClaraGrthns]

Description

There is an Import error in the pts module in gluonts/nursery/robust-mts-attack/pts/modules/distribution_output.py

To Reproduce

The following lines of code yield the Import Error:

from gluonts.torch.modules.distribution_output import (
    DistributionOutput,
    LambdaLayer,
    PtArgProj,
)

Error message or code output

Traceback (most recent call last):
  File "gluonts/src/gluonts/nursery/robust-mts-attack/pts/modules/distribution_output.py", line 47, in <module>
    from gluonts.torch.modules.distribution_output import (
ModuleNotFoundError: No module named 'gluonts.torch.modules.distribution_output'

Proposed Solution:

The following lines of code should solve the problem to match the current version of gluonts (v 0.15.0).

from gluonts.torch.distributions.distribution_output import DistributionOutput
from gluonts.torch.modules.lambda_layer import LambdaLayer
from gluonts.torch.distributions.output import PtArgProj

Environment

• Operating system: MacOS 14.0
• Python version: 3.11.
• GluonTS version: 0.15.0

[ucmcidt]

There is a similar Import error in the pts model "deepvar" in n gluonts/nursery/robust-mts-attack/pts/model/deepvar/ in the two files
deepvar_estimator.py and deepvar_network.py

e.g.:
robust-mts-attack\pts\model\deepvar\deepvar_network.py", line 22, in
from gluonts.torch.modules.distribution_output import DistributionOutput
ModuleNotFoundError: No module named 'gluonts.torch.modules.distribution_output'

It is solved by modifying the line:
from gluonts.torch.modules.distribution_output import DistributionOutput
by:
from gluonts.torch.distributions.distribution_output import DistributionOutput

to work on GluonTS v0.16.0

Operating system: Windows 11
Python version: 3.11.11
GluonTS version: 0.16.0