From 36164f6f4d7c1d416aeec0949d61aa2d9f77a5f4 Mon Sep 17 00:00:00 2001 From: Clara Grotehans <68223176+ClaraGrthns@users.noreply.github.com> Date: Fri, 14 Jun 2024 15:31:45 +0200 Subject: [PATCH] Update: Import of gluonts modules in pts module (#3194) *Issue #, if available:* #3193 *Description of changes:* Import Error solved with updated imports matching the new version of gluonts By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice. Tag: bug fix (import error) Co-authored-by: Clara Grotehans Co-authored-by: Lorenzo Stella --- .../robust-mts-attack/pts/modules/distribution_output.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/src/gluonts/nursery/robust-mts-attack/pts/modules/distribution_output.py b/src/gluonts/nursery/robust-mts-attack/pts/modules/distribution_output.py index 9b6a4cb157..dd528262af 100644 --- a/src/gluonts/nursery/robust-mts-attack/pts/modules/distribution_output.py +++ b/src/gluonts/nursery/robust-mts-attack/pts/modules/distribution_output.py @@ -44,11 +44,10 @@ TransformedImplicitQuantile, ) from gluonts.core.component import validated -from gluonts.torch.modules.distribution_output import ( - DistributionOutput, - LambdaLayer, - PtArgProj, -) +from gluonts.torch.distributions.distribution_output import DistributionOutput +from gluonts.torch.modules.lambda_layer import LambdaLayer +from gluonts.torch.distributions.output import PtArgProj + from pts.modules.iqn_modules import ImplicitQuantileModule