build(deps): packages

Author: kozistr

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pytorch_optimizer"
-version = "2.4.1"
+version = "2.4.2"
 description = "optimizer & lr scheduler implementations in PyTorch with clean-code, strict types. Also, including useful optimization ideas."
 license = "Apache-2.0"
 authors = ["kozistr <[email protected]>"]
@@ -34,10 +34,10 @@ classifiers = [
 [tool.poetry.dependencies]
 python = "^3.7.2"
 numpy = [
-    { version = "=1.21.1", python = ">=3.7,<3.8"},
-    { version = "*", python = ">=3.8"},
+    { version = "=1.21.1", python = ">=3.7,<3.8" },
+    { version = "*", python = ">=3.8" },
 ]
-torch = "^1.10"
+torch = { version = ">=1.10", source = "torch" }
 
 [tool.poetry.dev-dependencies]
 isort = "^5.11.5"
@@ -46,6 +46,11 @@ ruff = "^0.0.244"
 pytest = "^7.2.1"
 pytest-cov = "^4.0.0"
 
+[[tool.poetry.source]]
+name = "torch"
+url = "https://download.pytorch.org/whl/cpu"
+secondary = true
+
 [tool.ruff]
 select = ["A", "B", "C4", "D", "E", "F", "G", "I", "N", "S", "T", "ISC", "ICN", "W", "INP", "PIE", "T20", "RET", "SIM", "TID", "ARG", "ERA", "RUF", "YTT", "PL"]
 ignore = ["D100", "D102", "D104", "D105", "D107", "D203", "D213", "PIE790", "PLR0912", "PLR0913", "PLR0915", "PLR2004"]

pyproject.toml

@@ -1,3 +1,5 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+
 attrs==22.2.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 black==23.1.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 click==8.1.3 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
@@ -17,10 +19,8 @@ pluggy==1.0.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 pytest-cov==4.0.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 pytest==7.2.1 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 ruff==0.0.244 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
-setuptools==67.2.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0" and platform_system == "Linux"
 tomli==2.0.1 ; python_full_version >= "3.7.2" and python_full_version <= "3.11.0a6"
-torch==1.13.1 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
+torch==1.13.1+cpu ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 typed-ast==1.5.4 ; python_version < "3.8" and implementation_name == "cpython" and python_full_version >= "3.7.2"
 typing-extensions==4.4.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
-wheel==0.38.4 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0" and platform_system == "Linux"
 zipp==3.13.0 ; python_full_version >= "3.7.2" and python_version < "3.8"

requirements-dev.txt

@@ -1,6 +1,6 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+
 numpy==1.21.1 ; python_full_version >= "3.7.2" and python_version < "3.8"
 numpy==1.24.2 ; python_version >= "3.8" and python_full_version < "4.0.0"
-setuptools==67.2.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0" and platform_system == "Linux"
-torch==1.13.1 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
+torch==1.13.1+cpu ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 typing-extensions==4.4.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
-wheel==0.38.4 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0" and platform_system == "Linux"