Merge pull request #6 from normal-computing/ruff

Remove black and add CI

.github/workflows/tests.yaml

@@ -0,0 +1,37 @@
+name: Tests
+
+on:
+  pull_request:
+    branches: [main]
+  push:
+    branches: [main]
+
+jobs:
+  pre-commit:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v3
+    - uses: actions/setup-python@v3
+    - uses: pre-commit/action@v3.0.0
+
+  tests:
+    name: Run tests for Python ${{ matrix.python-version }}
+    runs-on: ubuntu-latest
+    needs:
+      - pre-commit
+    strategy:
+      matrix:
+        python-version: ['3.10']
+    steps:
+      - uses: actions/checkout@v3
+      - name: Set up Python ${{ matrix.python-version }}
+        uses: actions/setup-python@v3
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Install dev environment
+        run: |
+          python -m pip install --upgrade pip
+          pip install .[test]
+      - name: Run the tests with pytest
+        run: |
+          python -m pytest

.pre-commit-config.yaml

@@ -1,10 +1,7 @@
 repos:
-  - repo: https://github.com/psf/black
-    rev: 23.11.0
-    hooks:
-    - id: black
   - repo: https://github.com/astral-sh/ruff-pre-commit
     rev: v0.1.4
     hooks:
     - id: ruff
-      args: ["--fix"]
+      args: ["--fix"]
+    - id: ruff-format

examples/utils/utils.py

@@ -4,7 +4,7 @@ def parse_devices(devices):
     for device in devices:
         try:
             device = int(device)
-        except:
+        except ValueError:
             pass
         devices_list.append(device)
     return devices_list

pyproject.toml

@@ -16,6 +16,9 @@ classifiers = [
 ]
 dependencies = ["torch>=2.0.0", "torchopt>=0.7.3", "optree>=0.10.0"]
 
+[project.optional-dependencies]
+test = ["pre-commit", "pytest"]
+
 [tool.setuptools]
 packages = ["uqlib"]