address review comment

apache · Sep 4, 2019 · 2d718c1 · 2d718c1
1 parent bca8c2f
commit 2d718c1
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/python/unittest/test_profiler.py b/tests/python/unittest/test_profiler.py
@@ -108,8 +108,7 @@ def get_log():
     task.start()
     start = time.time()
     var = mx.nd.ones((1000, 500))
-    log = get_log()
-    assert len(log) == 10
+    assert len(get_log()) == 10
     var.asnumpy()
     stop = time.time()
     task.stop()
@@ -136,8 +135,7 @@ def get_log():
     frame.start()
     start = time.time()
     var = mx.nd.ones((1000, 500))
-    log = get_log()
-    assert len(log) == 100000
+    assert len(get_log()) == 100000
     var.asnumpy()
     stop = time.time()
     frame.stop()