From 0625052be27a7dbda2f80b73d0e66e33040b5fed Mon Sep 17 00:00:00 2001 From: co63oc Date: Tue, 25 Jul 2023 11:12:41 +0800 Subject: [PATCH] Fix --- tests/test_nn_Module_type.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_nn_Module_type.py b/tests/test_nn_Module_type.py index 742212dba..ecb0c400e 100644 --- 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 """ )