Merge pull request #53 from narumiruna/fix-ruff-workflow

fix ruff workflow

pyproject.toml

@@ -40,14 +40,15 @@ select = [
     "I", # isort
     # "UP", # pyupgrade
     "W", # pycodestyle warnings
-
 ]
 
 [tool.ruff.lint.per-file-ignores]
 "__init__.py" = ["F401", "F403"]
 
 [tool.ruff.lint.isort]
 force-single-line = true
+# https://github.com/astral-sh/ruff/issues/10519
+known-third-party = ["wandb"]
 
 [tool.pytest.ini_options]
 filterwarnings = ["ignore::DeprecationWarning"]

template/cli.py

@@ -1,9 +1,8 @@
 import click
+import wandb
 from mlconfig import instantiate
 from mlconfig import load
 
-import wandb
-
 
 @click.command()
 @click.option("-c", "--config-file", type=click.STRING, default="configs/mnist.yaml")

template/datasets/mnist.py

@@ -17,6 +17,4 @@ def __init__(self, root: str, train: bool, batch_size: int, **kwargs):
 
         dataset = datasets.MNIST(root, train=train, transform=transform, download=True)
 
-        super(MNISTDataLoader, self).__init__(
-            dataset=dataset, batch_size=batch_size, shuffle=train, **kwargs
-        )
+        super(MNISTDataLoader, self).__init__(dataset=dataset, batch_size=batch_size, shuffle=train, **kwargs)

template/trainers/mnist.py

@@ -1,5 +1,6 @@
 import torch
 import torch.nn.functional as F
+import wandb
 from mlconfig import register
 from torch.nn import Module
 from torch.optim import Optimizer
@@ -10,8 +11,6 @@
 from tqdm import tqdm
 from tqdm import trange
 
-import wandb
-
 from .trainer import Trainer