From de6a705a3aced440eaaf916c4025d4e40fb62fc7 Mon Sep 17 00:00:00 2001 From: Clara Grotehans Date: Fri, 14 Jun 2024 12:44:00 +0200 Subject: [PATCH] updated imports of gluonts modules in pts module --- .../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