style(mypy): silence incorrect typing warnings (#375)

* style(pyproject): update mypy config

* style(mypy): silenced typing warnings

holocron/optim/adabelief.py

@@ -8,7 +8,7 @@
 
 import torch
 from torch import Tensor
-from torch.optim import Adam
+from torch.optim import Adam  # type: ignore[attr-defined]
 
 __all__ = ["AdaBelief", "adabelief"]
 

holocron/optim/adamp.py

@@ -9,7 +9,7 @@
 import torch
 from torch import Tensor
 from torch.nn import functional as F
-from torch.optim import Adam
+from torch.optim import Adam  # type: ignore[attr-defined]
 
 __all__ = ["AdamP", "adamp"]
 

holocron/optim/adan.py

@@ -8,7 +8,7 @@
 
 import torch
 from torch import Tensor
-from torch.optim import Adam
+from torch.optim import Adam  # type: ignore[attr-defined]
 
 __all__ = ["Adan", "adan"]
 

holocron/optim/ademamix.py

@@ -8,7 +8,7 @@
 
 import torch
 from torch import Tensor
-from torch.optim import Optimizer
+from torch.optim import Optimizer  # type: ignore[attr-defined]
 
 __all__ = ["AdEMAMix", "ademamix"]
 

holocron/optim/wrapper.py

@@ -30,7 +30,7 @@ class Lookahead(Optimizer):
 
     def __init__(
         self,
-        base_optimizer: torch.optim.Optimizer,
+        base_optimizer: torch.optim.Optimizer,  # type: ignore[name-defined]
         sync_rate: float = 0.5,
         sync_period: int = 6,
     ) -> None:
@@ -154,7 +154,7 @@ class Scout(Optimizer):
 
     def __init__(
         self,
-        base_optimizer: torch.optim.Optimizer,
+        base_optimizer: torch.optim.Optimizer,  # type: ignore[name-defined]
         sync_rate: float = 0.5,
         sync_period: int = 6,
     ) -> None:

holocron/trainer/core.py

@@ -48,7 +48,7 @@ def __init__(
         train_loader: DataLoader,
         val_loader: DataLoader,
         criterion: nn.Module,
-        optimizer: torch.optim.Optimizer,
+        optimizer: torch.optim.Optimizer,  # type: ignore[name-defined]
         gpu: Optional[int] = None,
         output_file: str = "./checkpoint.pth",
         amp: bool = False,

pyproject.toml

@@ -204,7 +204,7 @@ quote-style = "double"
 indent-style = "space"
 
 [tool.mypy]
-python_version = "3.9"
+python_version = "3.11"
 files = "holocron/"
 show_error_codes = true
 pretty = true