From 54244425bbd407c33b0144e585ea92651926f75f Mon Sep 17 00:00:00 2001 From: Lorenzo Stella Date: Fri, 26 Aug 2022 13:43:13 +0200 Subject: [PATCH] compat fix after backport --- test/mx/test_no_batches.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/mx/test_no_batches.py b/test/mx/test_no_batches.py index 9423e6e1be..b874277327 100644 --- a/test/mx/test_no_batches.py +++ b/test/mx/test_no_batches.py @@ -14,7 +14,7 @@ import pytest from gluonts.exceptions import GluonTSDataError -from gluonts.mx.model.deepar import DeepAREstimator +from gluonts.model.deepar import DeepAREstimator from gluonts.mx.trainer import Trainer