Skip to content

Commit

Permalink
Re-format the inference utils use ruff pre-commit
Browse files Browse the repository at this point in the history
Re-format the inference utils use ruff pre-commit
  • Loading branch information
erizmr authored Oct 12, 2024
2 parents 51b1808 + 3ac928d commit 14be7f2
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion inference_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,6 +151,7 @@ def find_bd(mesh, function_space, use_4_edge=False, poly_mesh=False):

# return data


class InputPack:
def __init__(
self,
Expand All @@ -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)
Expand Down

0 comments on commit 14be7f2

Please sign in to comment.