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 15, 2024 · a6ccfa3 · a6ccfa3
1 parent 889a52d
commit a6ccfa3
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