isort

src/gluonts/torch/model/softs/__init__.py

@@ -11,9 +11,9 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-from .module import SofTSModel
-from .lightning_module import SofTSLightningModule
 from .estimator import SofTSEstimator
+from .lightning_module import SofTSLightningModule
+from .module import SofTSModel
 
 __all__ = [
     "SofTSModel",

src/gluonts/torch/model/softs/estimator.py

@@ -11,30 +11,30 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-from typing import Optional, Iterable, Dict, Any
+from typing import Any, Dict, Iterable, Optional
 
-import torch
 import lightning.pytorch as pl
+import torch
 
 from gluonts.core.component import validated
 from gluonts.dataset.common import Dataset
 from gluonts.dataset.field_names import FieldName
 from gluonts.dataset.loader import as_stacked_batches
 from gluonts.itertools import Cyclic
+from gluonts.torch.distributions import Output, StudentTOutput
+from gluonts.torch.model.estimator import PyTorchLightningEstimator
+from gluonts.torch.model.predictor import PyTorchPredictor
 from gluonts.transform import (
-    AsNumpyArray,
-    Transformation,
     AddObservedValuesIndicator,
+    AsNumpyArray,
+    ExpectedNumInstanceSampler,
     InstanceSampler,
     InstanceSplitter,
-    ValidationSplitSampler,
-    TestSplitSampler,
-    ExpectedNumInstanceSampler,
     SelectFields,
+    TestSplitSampler,
+    Transformation,
+    ValidationSplitSampler,
 )
-from gluonts.torch.model.estimator import PyTorchLightningEstimator
-from gluonts.torch.model.predictor import PyTorchPredictor
-from gluonts.torch.distributions import Output, StudentTOutput
 
 from .lightning_module import SofTSLightningModule
 

src/gluonts/torch/model/softs/module.py

@@ -20,7 +20,7 @@
 from gluonts.core.component import validated
 from gluonts.model import Input, InputSpec
 from gluonts.torch.distributions import StudentTOutput
-from gluonts.torch.scaler import StdScaler, MeanScaler, NOPScaler
+from gluonts.torch.scaler import MeanScaler, NOPScaler, StdScaler
 from gluonts.torch.util import weighted_average