From 0422769340e50fca33d437ab7e964f5a96cf91a8 Mon Sep 17 00:00:00 2001 From: Bartlomiej Gawrych Date: Tue, 30 Nov 2021 13:10:37 +0100 Subject: [PATCH] Fix sanity CI --- python/mxnet/context.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/mxnet/context.py b/python/mxnet/context.py index 47428c370ea7..ee34641162cf 100644 --- a/python/mxnet/context.py +++ b/python/mxnet/context.py @@ -16,8 +16,8 @@ # under the License. """Context management API of mxnet.""" from warnings import warn -from .device import (Device, _current, cpu, gpu, cpu_pinned, - num_gpus, gpu_memory_info) # pylint: disable=unused-import +from .device import Device, _current, cpu, gpu, cpu_pinned # pylint: disable=unused-import +from .device import num_gpus, gpu_memory_info # pylint: disable=unused-import def Context(*args, **kwargs):