Merge pull request #88 from kozistr/ci/test

[CI] Reduce `num_iterations` to speed up the testing

Author: kozistr

.github/workflows/ci.yml

@@ -23,10 +23,8 @@ jobs:
       uses: actions/cache@v3
       with:
         path: ~/.cache/pip
-        key: ${{ runner.os }}-pip-${{ hashFiles('requirements.txt') }}
-        restore-keys: |
-          ${{ runner.os }}-pip-
-          ${{ runner.os }}-
+        key: ${{ runner.os }}-pip-${{ hashFiles('requirements-dev.txt') }}
+        restore-keys: ${{ runner.os }}-pip-
     - name: Install dependencies
       run: pip install -r requirements-dev.txt
     - name: Check lint

.github/workflows/publish.yml

@@ -36,10 +36,8 @@ jobs:
         uses: actions/cache@v3
         with:
           path: ~/.cache/pip
-          key: ${{ runner.os }}-pip-${{ hashFiles('requirements.txt') }}
-          restore-keys: |
-            ${{ runner.os }}-pip-
-            ${{ runner.os }}-
+          key: ${{ runner.os }}-pip-${{ hashFiles('requirements-dev.txt') }}
+          restore-keys: ${{ runner.os }}-pip-
       - name: Install dependencies
         run: |
           python3 -m pip install poetry

README.rst

@@ -31,6 +31,12 @@ Install
 
     $ pip3 install -U pytorch-optimizer
 
+or
+
+::
+
+    $ pip3 install -U --no-deps pytorch-optimizer
+
 Simple Usage
 ~~~~~~~~~~~~
 

poetry.lock

@@ -37,7 +37,7 @@ numpy = [
     { 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.4"
@@ -46,6 +46,11 @@ pylint = "^2.15.9"
 pytest = "^7.2.0"
 pytest-cov = "^4.0.0"
 
+[[tool.poetry.source]]
+name = "torch"
+url = "https://download.pytorch.org/whl/cpu"
+secondary = true
+
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"

pyproject.toml

@@ -1,3 +1,5 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
+
 astroid==2.12.13 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 attrs==22.2.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 black==22.12.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
@@ -16,17 +18,15 @@ numpy==1.21.1 ; python_full_version >= "3.7.2" and python_version < "3.8"
 numpy==1.24.1 ; python_version >= "3.8" and python_full_version < "4.0.0"
 packaging==22.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 pathspec==0.10.3 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
-platformdirs==2.6.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
+platformdirs==2.6.2 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 pluggy==1.0.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 pylint==2.15.9 ; 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.0 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
-setuptools==65.6.3 ; 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.0a7"
 tomlkit==0.11.6 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
-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"
 wrapt==1.14.1 ; python_full_version >= "3.7.2" and python_full_version < "4.0.0"
 zipp==3.11.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.1 ; python_version >= "3.8" and python_full_version < "4.0.0"
-setuptools==65.6.3 ; 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"