update imports

This commit is contained in:
Kashif Rasul
2021-12-12 14:06:40 +01:00
parent 0b0dfe12ca
commit 0c4b37a48c
+1 -1
View File
@@ -14,7 +14,7 @@ from gluonts.dataset.common import Dataset
from gluonts.model.estimator import Estimator
from gluonts.torch.model.predictor import PyTorchPredictor
from gluonts.transform import SelectFields, Transformation
from gluonts.support.util import maybe_len
from gluonts.itertools import maybe_len
from pts import Trainer
from pts.model import get_module_forward_input_names