From e3f42c8aa01d88cb32b1c59adaadfe11ee1034e6 Mon Sep 17 00:00:00 2001
From: shaoanlu <shaoanlu@gmail.com>
Date: Sun, 28 Apr 2024 14:43:16 +0900
Subject: [PATCH] Create unittest workflow

---
 .github/workflows/unittest.yml | 38 ++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 .github/workflows/unittest.yml

diff --git a/.github/workflows/unittest.yml b/.github/workflows/unittest.yml
new file mode 100644
index 0000000..9155fca
--- /dev/null
+++ b/.github/workflows/unittest.yml
@@ -0,0 +1,38 @@
+name: Python Unit Tests
+
+# Defines when the action should run. This example triggers on push and pull requests to the main branch.
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+
+# Jobs that the workflow will execute
+jobs:
+  run-unittests:
+    # The type of runner that the job will run on
+    runs-on: ubuntu-latest
+    strategy:
+      matrix:
+        python-version: ["3.10"]
+
+    # Steps represent a sequence of tasks that will be executed as part of the job
+    steps:
+    # Checks-out your repository under $GITHUB_WORKSPACE, so your job can access it
+    - uses: actions/checkout@v3
+
+    # Sets up a Python environment
+    - name: Set up Python ${{ matrix.python-version }}
+      uses: actions/setup-python@v4
+      with:
+        python-version: ${{ matrix.python-version }}
+
+    # Install dependencies (if any)
+    - name: Install dependencies
+      run: |
+        python -m pip install --upgrade pip
+        pip install torch==1.13.1 torchvision==0.14.1 jax==0.4.23 jaxlib==0.4.23 diffusers==0.18.2 joblib
+
+    # Run unittests using the Python unittest module
+    - name: Run unittest
+      run: python -m unittest discover