fix undefined-variable-1

python/paddle/fluid/net_drawer.py

@@ -102,11 +102,11 @@ def parse_graph(program, graph, var_dict, **kwargs):
 
 def draw_graph(startup_program, main_program, **kwargs):
     if "graph_attr" in kwargs:
-        GRAPH_STYLE.update(kwargs[graph_attr])
+        GRAPH_STYLE.update(kwargs["graph_attr"])
     if "node_attr" in kwargs:
-        OP_STYLE.update(kwargs[node_attr])
+        OP_STYLE.update(kwargs["node_attr"])
     if "edge_attr" in kwargs:
-        VAR_STYLE.update(kwargs[edge_attr])
+        VAR_STYLE.update(kwargs["edge_attr"])
 
     graph_id = unique_id()
     filename = kwargs.get("filename")

python/paddle/fluid/tests/unittests/dygraph_to_static/simnet_dygraph_model.py

@@ -18,6 +18,7 @@
 from functools import reduce
 from paddle.fluid.dygraph import declarative, to_variable
 from paddle.fluid.dygraph import Embedding, Layer, Linear
+from paddle.static import Variable
 
 
 class EmbeddingLayer(object):

python/paddle/fluid/tests/unittests/dygraph_to_static/simnet_dygraph_model_v2.py

@@ -14,6 +14,7 @@
 
 from functools import reduce
 import paddle
+from paddle.static import Variable
 
 
 class EmbeddingLayer(object):

python/paddle/fluid/tests/unittests/dygraph_to_static/test_static_analysis.py

@@ -17,6 +17,7 @@
 import gast
 import inspect
 import numpy as np
+import paddle
 import paddle.fluid as fluid
 import unittest
 

python/paddle/fluid/tests/unittests/mkldnn/test_conv2d_transpose_mkldnn_op.py

@@ -18,6 +18,7 @@
 import numpy as np
 import paddle.fluid.core as core
 from paddle.fluid.tests.unittests.op_test import OpTest
+from paddle import enable_static
 
 from paddle.fluid.tests.unittests.test_conv2d_transpose_op import conv2dtranspose_forward_naive, TestConv2DTransposeOp
 

python/paddle/text/datasets/wmt14.py

@@ -17,6 +17,7 @@
 import tarfile
 import numpy as np
 import gzip
+import six
 
 from paddle.io import Dataset
 import paddle.compat as cpt