[Dy2St] Fix test_pylayer random failed (#58515)

PaddlePaddle · Oct 31, 2023 · 75afa68 · 75afa68
1 parent f0ea1d6
commit 75afa68
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/test/dygraph_to_static/test_pylayer.py b/test/dygraph_to_static/test_pylayer.py
@@ -26,7 +26,6 @@
 import unittest
 
 import numpy as np
-from dygraph_to_static_utils_new import Dy2StTestBase
 from test_jit_save_load import train
 
 import paddle
@@ -263,7 +262,7 @@ def forward(self, x):
         return out
 
 
-class TestPyLayerBase(Dy2StTestBase):
+class TestPyLayerBase(unittest.TestCase):
     def setUp(self):
         self.place = "gpu" if paddle.is_compiled_with_cuda() else "cpu"
         self.to_static = False
@@ -513,7 +512,7 @@ def test_pylayer_net_with_no_grad(self):
         self._run_and_compare(input1, input2)
 
 
-class PyLayerTrainHelper(Dy2StTestBase):
+class PyLayerTrainHelper(unittest.TestCase):
     def setUp(self):
         self.place = "gpu" if paddle.is_compiled_with_cuda() else "cpu"
 
@@ -586,7 +585,7 @@ def test_pylayer_net_no_grad(self):
         )
 
 
-class TestPyLayerJitSaveLoad(Dy2StTestBase):
+class TestPyLayerJitSaveLoad(unittest.TestCase):
     def setUp(self):
         self.temp_dir = tempfile.TemporaryDirectory()
         self.model_path = os.path.join(