From 0c4b37a48cf02f6e567e8f5c150cd5ea91a34c15 Mon Sep 17 00:00:00 2001
From: Kashif Rasul <kashif.rasul@gmail.com>
Date: Sun, 12 Dec 2021 14:06:40 +0100
Subject: [PATCH] update imports

---
 pts/model/estimator.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pts/model/estimator.py b/pts/model/estimator.py
index 0359683..64968f3 100644
--- a/pts/model/estimator.py
+++ b/pts/model/estimator.py
@@ -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