From 83284c8cf8a32311dc0844896bd4d32e50b42dff Mon Sep 17 00:00:00 2001
From: Jiangxinz <jiangxinz@foxmail.com>
Date: Mon, 28 Jun 2021 19:46:27 +0800
Subject: [PATCH] fix undef var (#33780)

---
 .../distributed/fleet/meta_optimizers/ascend/ascend_parser.py   | 2 +-
 .../fluid/tests/unittests/asp/test_asp_pruning_2d_best.py       | 1 +
 .../fluid/tests/unittests/npu/test_collective_base_npu.py       | 1 +
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_parser.py b/python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_parser.py
index 3331a45b3d9..5f2b6df493b 100644
--- a/python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_parser.py
+++ b/python/paddle/distributed/fleet/meta_optimizers/ascend/ascend_parser.py
@@ -136,7 +136,7 @@ class AscendHelper(object):
 
     def dtype2np(self, index):
         assert index in self.dtype2np_map, "index[%d] is not supported %d" % (
-            dtype)
+            index)
         return self.dtype2np_map[index]
 
 
diff --git a/python/paddle/fluid/tests/unittests/asp/test_asp_pruning_2d_best.py b/python/paddle/fluid/tests/unittests/asp/test_asp_pruning_2d_best.py
index 1b8b1e4a06a..b21f8edf4f4 100644
--- a/python/paddle/fluid/tests/unittests/asp/test_asp_pruning_2d_best.py
+++ b/python/paddle/fluid/tests/unittests/asp/test_asp_pruning_2d_best.py
@@ -16,6 +16,7 @@
 from __future__ import print_function
 
 import paddle
+import unittest
 from paddle.fluid.contrib import sparsity
 from paddle.fluid.tests.unittests.asp.asp_pruning_base import TestASPHelperPruningBase
 
diff --git a/python/paddle/fluid/tests/unittests/npu/test_collective_base_npu.py b/python/paddle/fluid/tests/unittests/npu/test_collective_base_npu.py
index b871256acd4..6372e1ab85f 100644
--- a/python/paddle/fluid/tests/unittests/npu/test_collective_base_npu.py
+++ b/python/paddle/fluid/tests/unittests/npu/test_collective_base_npu.py
@@ -27,6 +27,7 @@ from contextlib import closing
 import paddle.fluid as fluid
 import paddle.fluid.unique_name as nameGen
 from paddle.fluid import core
+from six import string_types
 
 
 class TestCollectiveRunnerBase(object):
-- 
GitLab