diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml new file mode 100644 index 00000000..0ebfbb6f --- /dev/null +++ b/.github/workflows/ci.yaml @@ -0,0 +1,22 @@ +name: Python package + +on: [push, pull_request] + +jobs: + build: + + runs-on: ubuntu-latest + strategy: + matrix: + python-version: [3.9] + + steps: + - uses: actions/checkout@v2 + - name: Set up Python ${{ matrix.python-version }} + uses: actions/setup-python@v2 + with: + python-version: ${{ matrix.python-version }} + - name: Install + run: pip install . + - name: Test imports work + run: python scripts/test_import.py diff --git a/scripts/test_import.py b/scripts/test_import.py new file mode 100644 index 00000000..aef93bef --- /dev/null +++ b/scripts/test_import.py @@ -0,0 +1,3 @@ +# A simple test to ensure that packages import correctly +import taming.models.cond_transformer +import taming.models.vqgan