From 3ac928d99dfa08bb5b4c6156f0e1db88850569b7 Mon Sep 17 00:00:00 2001 From: mingrui Date: Sun, 13 Oct 2024 03:02:15 +0800 Subject: [PATCH] fix the linter --- inference_utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/inference_utils.py b/inference_utils.py index fb38eae..01e0186 100644 --- a/inference_utils.py +++ b/inference_utils.py @@ -151,6 +151,7 @@ def find_bd(mesh, function_space, use_4_edge=False, poly_mesh=False): # return data + class InputPack: def __init__( self, @@ -176,7 +177,7 @@ def __init__( .float() .to(device) ) - + self.edge_index = torch.tensor(edge_index).to(torch.int64).to(device) self.bd_mask = torch.tensor(bd_mask).reshape(-1, 1).to(device) self.node_num = torch.tensor(self.coord.shape[0]).to(device)