Python2 doesn't scope list comprehensions

apache · Apr 26, 2019 · 25fda1d · 25fda1d
1 parent afa41c7
commit 25fda1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -924,7 +924,7 @@ def test_dense_backward():
     with net.name_scope():
         net.add(gluon.nn.Dense(1, in_units=x.shape[1]))
     net.initialize(mx.initializer.Constant(.5))
-    params = [x.data() for x in net.collect_params().values()]
+    params = [p.data() for p in net.collect_params().values()]
     x.attach_grad()
     with ag.record():
         y = net.forward(x)