Correct codacy warnings

This commit is contained in:
FilippoOlivo
2024-10-22 14:26:39 +02:00
committed by Nicola Demo
parent c9304fb9bb
commit 1bc1b3a580
15 changed files with 252 additions and 210 deletions

View File

@@ -1,20 +1,18 @@
import torch
import pytest
from pina import TorchOptimizer
opt_list = [
torch.optim.Adam,
torch.optim.AdamW,
torch.optim.SGD,
torch.optim.RMSprop
torch.optim.Adam, torch.optim.AdamW, torch.optim.SGD, torch.optim.RMSprop
]
@pytest.mark.parametrize("optimizer_class", opt_list)
def test_constructor(optimizer_class):
TorchOptimizer(optimizer_class, lr=1e-3)
@pytest.mark.parametrize("optimizer_class", opt_list)
def test_hook(optimizer_class):
opt = TorchOptimizer(optimizer_class, lr=1e-3)
opt.hook(torch.nn.Linear(10, 10).parameters())
opt.hook(torch.nn.Linear(10, 10).parameters())