Skip to content

Fix Efficientnet enocoder for torchscript (#1037) #732

Fix Efficientnet enocoder for torchscript (#1037)

Fix Efficientnet enocoder for torchscript (#1037) #732

Workflow file for this run

# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions
name: CI
on:
push:
branches: [ main ]
pull_request:
branches: [ main ]
jobs:
style:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- name: Set up Python
uses: astral-sh/setup-uv@v5
with:
python-version: "3.11"
- name: Install dependencies
run: uv pip install -r requirements/test.txt
# Update output format to enable automatic inline annotations.
- name: Run Ruff Linter
run: ruff check --output-format=github
- name: Run Ruff Formatter
run: ruff format --check
test:
strategy:
matrix:
os: [ubuntu-latest, macos-latest, windows-latest]
python-version: ["3.10", "3.11", "3.12"]
runs-on: ${{ matrix.os }}
steps:
- uses: actions/checkout@v4
- name: Set up Python ${{ matrix.python-version }}
uses: astral-sh/setup-uv@v5
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: uv pip install -r requirements/required.txt -r requirements/test.txt
- name: Show installed packages
run: uv pip list
- name: Test with PyTest
run: uv run pytest -v -rsx -n 2 --cov=segmentation_models_pytorch --cov-report=xml --cov-config=pyproject.toml --non-marked-only
- name: Upload coverage reports to Codecov
uses: codecov/codecov-action@v5
with:
token: ${{ secrets.CODECOV_TOKEN }}
slug: qubvel-org/segmentation_models.pytorch
if: matrix.os == 'macos-latest' && matrix.python-version == '3.12'
test_logits_match:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- name: Set up Python
uses: astral-sh/setup-uv@v5
with:
python-version: "3.10"
- name: Install dependencies
run: uv pip install -r requirements/required.txt -r requirements/test.txt
- name: Show installed packages
run: uv pip list
- name: Test with PyTest
run: RUN_SLOW=1 uv run pytest -v -rsx -n 2 -m "logits_match"
test_torch_compile:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- name: Set up Python
uses: astral-sh/setup-uv@v5
with:
python-version: "3.10"
- name: Install dependencies
run: uv pip install -r requirements/required.txt -r requirements/test.txt
- name: Show installed packages
run: uv pip list
- name: Test with PyTest
run: uv run pytest -v -rsx -n 2 -m "compile"
test_torch_export:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- name: Set up Python
uses: astral-sh/setup-uv@v5
with:
python-version: "3.10"
- name: Install dependencies
run: uv pip install -r requirements/required.txt -r requirements/test.txt
- name: Show installed packages
run: uv pip list
- name: Test with PyTest
run: uv run pytest -v -rsx -n 2 -m "torch_export"
test_torch_script:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- name: Set up Python
uses: astral-sh/setup-uv@v5
with:
python-version: "3.10"
- name: Install dependencies
run: uv pip install -r requirements/required.txt -r requirements/test.txt
- name: Show installed packages
run: uv pip list
- name: Test with PyTest
run: uv run pytest -v -rsx -n 2 -m "torch_script"
minimum:
runs-on: ubuntu-latest
steps:
- uses: actions/checkout@v4
- name: Set up Python
uses: astral-sh/setup-uv@v5
with:
python-version: "3.9"
- name: Install dependencies
run: uv pip install -r requirements/minimum.old -r requirements/test.txt
- name: Show installed packages
run: uv pip list
- name: Test with pytest
run: uv run pytest -v -rsx -n 2 --non-marked-only