Add GitHub action to test import after pip install

CompVis · Sep 28, 2021 · fefe15a · fefe15a
1 parent ee37e02
commit fefe15a
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/.github/workflows/ci.yaml 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
@@ -0,0 +1,3 @@
+# A simple test to ensure that packages import correctly
+import taming.models.cond_transformer
+import taming.models.vqgan