Skip to content

Commit 057aa84

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 05a59c9 commit 057aa84

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

‎sage/utils.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,8 @@ def model_conversion(model):
3636

3737
model.eval()
3838
device = next(model.parameters()).device
39-
return (
40-
lambda x: model(torch.tensor(x, dtype=torch.float32, device=device))
39+
return lambda x: (
40+
model(torch.tensor(x, dtype=torch.float32, device=device))
4141
.cpu()
4242
.data.numpy()
4343
)

0 commit comments

Comments
 (0)