Update: Import of gluonts modules in pts module (awslabs#3194)

*Issue #, if available:* awslabs#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 <ClaraGrthns> Co-authored-by: Lorenzo Stella <[email protected]>
kashif · Jun 14, 2024 · 36164f6 · 36164f6
1 parent b97ff43
commit 36164f6
Showing 1 changed file with 4 additions and 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
@@ -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