fix to_static import

Author: Aurelius84

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

@@ -17,7 +17,7 @@
 import paddle.fluid.param_attr as attr
 
 from functools import reduce
-from paddlle.jit.api import declarative
+from paddle.jit.api import declarative
 from paddle.fluid.dygraph import Embedding, Layer, Linear
 from paddle.static import Variable
 

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

@@ -16,10 +16,10 @@
 import paddle
 import paddle.fluid as fluid
 import unittest
-from paddle.jit.api import declarative
+from paddle.jit import to_static
 
 
-@fluid.dygraph.declarative
+@paddle.jit.to_static
 def dygraph_decorated_func(x):
     x = fluid.dygraph.to_variable(x)
     if paddle.mean(x) > 0:
@@ -29,7 +29,7 @@ def dygraph_decorated_func(x):
     return x_v
 
 
-@fluid.dygraph.declarative
+@paddle.jit.to_static
 def jit_decorated_func(x):
     x = fluid.dygraph.to_variable(x)
     if paddle.mean(x) > 0:
@@ -39,19 +39,19 @@ def jit_decorated_func(x):
     return x_v
 
 
-@fluid.dygraph.declarative
+@paddle.jit.to_static
 def decorated_call_decorated(x):
     return jit_decorated_func(x)
 
 
 class DoubleDecorated:
     @classmethod
-    @declarative
+    @to_static
     def double_decorated_func1(self, x):
         return dygraph_decorated_func(x)
 
     @classmethod
-    @fluid.dygraph.declarative
+    @paddle.jit.to_static
     def double_decorated_func2(self, x):
         return jit_decorated_func(x)
 
@@ -78,10 +78,8 @@ def test_run_success(self):
 
 class TestImportProgramTranslator(unittest.TestCase):
     def test_diff_pkg_same_cls(self):
-        dygraph_prog_trans = fluid.dygraph.ProgramTranslator()
-        dy_to_stat_prog_trans = (
-            fluid.dygraph.dygraph_to_static.ProgramTranslator()
-        )
+        dygraph_prog_trans = paddle.jit.ProgramTranslator()
+        dy_to_stat_prog_trans = paddle.jit.ProgramTranslator()
         full_pkg_prog_trans = paddle.jit.ProgramTranslator()
         self.assertEqual(dygraph_prog_trans, dy_to_stat_prog_trans)
         self.assertEqual(dygraph_prog_trans, full_pkg_prog_trans)

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

@@ -19,7 +19,8 @@
 import paddle.fluid as fluid
 from paddle.fluid.dygraph.nn import Linear, Embedding
 from paddle.fluid.dygraph import to_variable
-from paddle.jit import ProgramTranslator, declarative
+from paddle.jit import ProgramTranslator
+from paddle.jit.api import declarative
 
 from test_lac import DynamicGRU