Upgrade to Ruff v0.9.4

Author: TylerYep

.pre-commit-config.yaml

@@ -2,7 +2,7 @@ ci:
   skip: [mypy, pytest]
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.6.8
+    rev: v0.9.4
     hooks:
       - id: ruff
         args: [--fix]

ai_toolkit/metrics/accuracy.py

@@ -7,7 +7,7 @@
 
 class Accuracy(Metric):
     def __repr__(self) -> str:
-        return f"{self.name}: {100. * self.value:.2f}%"
+        return f"{self.name}: {100.0 * self.value:.2f}%"
 
     @staticmethod
     def calculate_accuracy(output: torch.Tensor, target: torch.Tensor) -> float:

ai_toolkit/models/__init__.py

@@ -24,9 +24,9 @@ def get_model_initializer(model_name: str) -> type[nn.Module]:
 
 __all__ = (
     "BasicCNN",
-    "DenseNet",
     "BasicLSTM",
-    "MaskRCNN",
     "BasicRNN",
+    "DenseNet",
+    "MaskRCNN",
     "get_model_initializer",
 )

ai_toolkit/test.py

@@ -46,7 +46,7 @@ def test_model(
     test_loss /= test_len
     print(
         f"\nTest set: Average loss: {test_loss:.4f},",
-        f"Accuracy: {correct}/{test_len} ({100. * correct / test_len:.2f}%)\n",
+        f"Accuracy: {correct}/{test_len} ({100.0 * correct / test_len:.2f}%)\n",
     )
 
 

ai_toolkit/train.py

@@ -86,9 +86,7 @@ def get_optimizer(args: Arguments, model: nn.Module) -> Optimizer:
     return AdamW(params, lr=args.lr)
 
 
-def get_scheduler(
-    args: Arguments, optimizer: Optimizer
-) -> lr_scheduler.LRScheduler:
+def get_scheduler(args: Arguments, optimizer: Optimizer) -> lr_scheduler.LRScheduler:
     return lr_scheduler.StepLR(optimizer, step_size=1, gamma=args.gamma)
 
 

ruff.toml

@@ -1,8 +1,6 @@
 target-version = "py312"
 lint.select = ["ALL"]
 lint.ignore = [
-	"ANN101",  # Missing type annotation for `self` in method
-	"ANN102",  # Missing type annotation for `cls` in classmethod
 	"ANN401",  # Dynamically typed expressions (typing.Any) are disallowed
 	"C901",    # function is too complex (12 > 10)
 	"COM812",  # Trailing comma missing