[Fix] type-hint (#404)

* fix: type-hint

* docs: v3.6.2 changelog

* build(deps): dev dependencies

* fix: type hint

* docs: v3.7.0 changelog

* update: test_version_utils

* fix: test_version_utils

* docs: v3.7.0 changelog

Author: kozistr

docs/changelogs/v3.6.2.md renamed to docs/changelogs/v3.7.0.md

@@ -9,6 +9,11 @@
 * Implement `EmoNavi`, `EmoFact`, and `EmoLynx` optimizers. (#393, #400)
     * [An emotion-driven optimizer that feels loss and navigates accordingly](https://github.com/muooon/EmoNavi)
 
+### CI
+
+* Enable CI for Python 3.8 ~ 3.13. (#402, #404)
+
 ### Fix
 
 * Adjust the value of `eps` to the fixed value `1e-15` when adding to `exp_avg_sq`. (#397, #398)
+* built-in type-hint in `Kron` optimizer. (#404)

poetry.lock

@@ -65,9 +65,12 @@ black = [
     { version = "<25", python = "<3.9" },
     { version = "^25", python = ">=3.9" },
 ]
-ruff = "*"
-pytest = "*"
-pytest-cov = "*"
+ruff = "^0.12"
+pytest = "^8"
+pytest-cov = [
+    { version = "<6", python = "<3.9" },
+    { version = "^6", python = ">=3.9" },
+]
 
 [[tool.poetry.source]]
 name = "torch"
@@ -139,5 +142,5 @@ omit = [
 ]
 
 [build-system]
-requires = ["poetry-core>=1.4.0"]
+requires = ["poetry-core>=2.0.0"]
 build-backend = "poetry.core.masonry.api"

pyproject.toml

@@ -222,7 +222,7 @@ def initialize_q_expressions(
     dtype: torch.dtype = dtype if dtype is not None else t.dtype
     shape = t.shape
     if len(shape) == 0:
-        qs: list[torch.Tensor] = [scale * torch.ones_like(t, dtype=dtype)]
+        qs: List[torch.Tensor] = [scale * torch.ones_like(t, dtype=dtype)]
         expressions_a: str = ',->'
         expression_gr: List[str] = [',->']
         expression_r: str = ',,->'
@@ -370,6 +370,6 @@ def update_precondition(
         q.sub_(tmp)
 
 
-def get_precondition_grad(qs: list[torch.Tensor], expressions: list[str], g: torch.Tensor) -> torch.Tensor:
+def get_precondition_grad(qs: List[torch.Tensor], expressions: List[str], g: torch.Tensor) -> torch.Tensor:
     r"""Precondition gradient G with pre-conditioner Q."""
     return torch.einsum(expressions[-1], *[x.conj() for x in qs], *qs, g)

pytorch_optimizer/optimizer/psgd.py

@@ -1,3 +1,5 @@
+from typing import Tuple
+
 import torch
 
 from pytorch_optimizer.base.exception import NoComplexParameterError, NoSparseGradientError
@@ -122,7 +124,7 @@ def init_group(self, group: GROUP, **kwargs) -> None:
                     ]
 
     @staticmethod
-    def get_scaled_lr(shape: tuple[int, int], lr: float, nonstandard_constant: float, max_dim: int = 10000) -> float:
+    def get_scaled_lr(shape: Tuple[int, int], lr: float, nonstandard_constant: float, max_dim: int = 10000) -> float:
         scale: float = (
             nonstandard_constant
             if len(shape) != 2 or shape[0] > max_dim or shape[1] > max_dim

pytorch_optimizer/optimizer/splus.py

@@ -4,7 +4,7 @@ black==24.8.0 ; python_version == "3.8"
 black==25.1.0 ; python_version >= "3.9"
 click==8.1.8 ; python_version >= "3.8"
 colorama==0.4.6 ; python_version >= "3.8" and (sys_platform == "win32" or platform_system == "Windows")
-coverage[toml]==7.10.0 ; python_version >= "3.9"
+coverage[toml]==7.10.1 ; python_version >= "3.9"
 coverage[toml]==7.6.1 ; python_version == "3.8"
 exceptiongroup==1.3.0 ; python_version < "3.11" and python_version >= "3.8"
 filelock==3.16.1 ; python_version == "3.8"
@@ -29,7 +29,8 @@ platformdirs==4.3.6 ; python_version == "3.8"
 platformdirs==4.3.8 ; python_version >= "3.9"
 pluggy==1.5.0 ; python_version == "3.8"
 pluggy==1.6.0 ; python_version >= "3.9"
-pytest-cov==5.0.0 ; python_version >= "3.8"
+pytest-cov==5.0.0 ; python_version == "3.8"
+pytest-cov==6.2.1 ; python_version >= "3.9"
 pytest==8.3.5 ; python_version >= "3.8"
 ruff==0.12.5 ; python_version >= "3.8"
 setuptools==80.9.0 ; python_version >= "3.12"

requirements-dev.txt

@@ -1,3 +1,4 @@
+import sys
 from typing import List
 
 import numpy as np
@@ -250,7 +251,12 @@ def test_version_utils():
     with pytest.raises(ValueError):
         parse_pytorch_version('a.s.d.f')
 
-    assert parse_pytorch_version(torch.__version__) == [2, 7, 1]
+    python_version = sys.version_info
+
+    if python_version.minor < 9:
+        assert parse_pytorch_version(torch.__version__) == [2, 4, 1]
+    else:
+        assert parse_pytorch_version(torch.__version__) == [2, 7, 1]
 
     assert compare_versions('2.7.0', '2.4.0') >= 0