From 1b778a64beb08acfc94eb0c4ac6f9c70b8e22d5f Mon Sep 17 00:00:00 2001
From: Lai Wei <royweilai@gmail.com>
Date: Fri, 29 Jun 2018 09:55:34 -0700
Subject: [PATCH] optimize imports

---
 tests/python/unittest/test_viz.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/tests/python/unittest/test_viz.py b/tests/python/unittest/test_viz.py
index 48fea71fb38c..eb5921f2e823 100644
--- a/tests/python/unittest/test_viz.py
+++ b/tests/python/unittest/test_viz.py
@@ -15,9 +15,11 @@
 # specific language governing permissions and limitations
 # under the License.
 
-import mxnet as mx
-import warnings
 import unittest
+import warnings
+
+import mxnet as mx
+
 
 def test_print_summary():
     data = mx.sym.Variable('data')
@@ -56,7 +58,7 @@ def test_plot_network():
     assert len(w) == 1
     assert "There are multiple variables with the same name in your graph" in str(w[-1].message)
     assert "fc" in str(w[-1].message)
-    
+
 if __name__ == "__main__":
     import nose
     nose.runmodule()