Merge pull request #8 from sensein/dev

distributed tests
sensein · Sep 18, 2024 · 1605391 · 1605391
2 parents bdb7e38 + b191fd0
commit 1605391
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 11 deletions.
diff --git a/.github/workflows/macos_tests.yaml b/.github/workflows/macos_tests.yaml
@@ -31,10 +31,5 @@ jobs:
           poetry install --no-interaction --no-root
         shell: bash
       - name: Run unit tests
-        run: poetry run pytest
-      - name: Upload Test Results
-        if: always()
-        uses: actions/upload-artifact@v4
-        with:
-          name: test-results
-          path: ./test-results
+        run: poetry run pytest -n auto
+        shell: bash
diff --git a/.github/workflows/ubuntu_tests_310.yaml b/.github/workflows/ubuntu_tests_310.yaml
@@ -76,7 +76,7 @@ jobs:
           poetry run nvidia-smi -L
           poetry run nvidia-smi -q -d Memory
       - name: Run unit tests
-        run: poetry run pytest
+        run: poetry run pytest -n auto
         shell: bash
       - name: Delete poetry env with python {{ matrix.python-version }}
         run: |

diff --git a/.github/workflows/ubuntu_tests_311.yaml b/.github/workflows/ubuntu_tests_311.yaml
@@ -76,7 +76,7 @@ jobs:
           poetry run nvidia-smi -L
           poetry run nvidia-smi -q -d Memory
       - name: Run unit tests
-        run: poetry run pytest
+        run: poetry run pytest -n auto
         shell: bash
       - name: Delete poetry env with python {{ matrix.python-version }}
         run: |

diff --git a/pyproject.toml b/pyproject.toml
@@ -8,8 +8,7 @@ readme = "README.md"
 [tool.poetry.dependencies]
 python = "^3.10"
 torch = "^2.4.1"
-pytest = "^8.3.3"
-
+pytest-xdist = {version = "^3.6.1", extras = ["psutil"]}
 
 [build-system]
 requires = ["poetry-core"]