Fix

PaddlePaddle · Jul 25, 2023 · 0625052 · 0625052
1 parent 111928e
commit 0625052
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_nn_Module_type.py b/tests/test_nn_Module_type.py
@@ -26,7 +26,7 @@ def test_case_1():
         x = torch.tensor([1., 2., 3.])
         module1 = torch.nn.Module()
         module1.register_buffer('buffer', x)
-        module1.type(torch.float32)
+        module1.type_unsupport(torch.float32)
         result = module1.buffer
         """
     )
@@ -45,7 +45,7 @@ def test_case_2():
         x = torch.tensor([1., 2., 3.])
         module1 = torch.nn.Module()
         module1.register_buffer('buffer', x)
-        module1.type(dst_type=torch.float32)
+        module1.type_unsupport(dst_type=torch.float32)
         result = module1.buffer
         """
     )