[ci skip] nd -> param

apache · Sep 7, 2020 · 2ec4357 · 2ec4357
1 parent 441f3bd
commit 2ec4357
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/python/gpu/test_profiler_gpu.py b/tests/python/gpu/test_profiler_gpu.py
@@ -164,10 +164,10 @@ def test_gpu_memory_profiler_gluon():
         # TODO: Remove this print statement later on.
         for row in csv_reader:
             print(",".join(list(row.values())))
-        for nd in model.collect_params().values():
-            expected_arg_name = "%s:in_arg:" % nd.var().attr('__profiler_scope__') + \
-                                nd.name
-            expected_arg_size = str(4 * np.prod(nd.shape))
+        for param in model.collect_params().values():
+            expected_arg_name = "%sin_arg:" % param.var().attr('__profiler_scope__') + \
+                                param.name
+            expected_arg_size = str(4 * np.prod(param.shape))
             csv_file.seek(0)
             entry_found = False
             for row in csv_reader: