ci.yml

name: CI for torchviz
 
 on:
   push:
     branches:
       - ci
   pull_request:
     branches:
       - ci
 
 jobs:
   test:
     name: Test on Python ${{ matrix.python-version }}
     runs-on: ubuntu-latest
 
     strategy:
       matrix:
         python-version: [3.8, 3.9, 3.10, 3.11]
 
     steps:
       - name: Checkout code
         uses: actions/checkout@v3
 
       - name: Set up Python
         uses: actions/setup-python@v4
         with:
           python-version: ${{ matrix.python-version }}
 
       - name: Install system dependencies
         run: |
           sudo apt-get update
           sudo apt-get install -y graphviz
 
       - name: Install Python dependencies
         run: |
           python -m pip install --upgrade pip
           pip install .
           pip install pytest
 
       - name: Run tests
         run: |
           pytest test/