Merge pull request #40 from will-rice/wr-refactor

Major Refactor

.github/workflows/python-package.yml

@@ -15,22 +15,18 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.10", "3.11", "3.12"]
 
     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 dependencies
-        run: |
-          python -m pip install --upgrade pip
-          python -m pip install ruff pytest
-          if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
-      - name: Lint with ruff
+
+      - name: Install uv
         run: |
-          ruff check .
-      - name: Test with pytest
+          curl -LsSf https://astral.sh/uv/install.sh | sh
+      - name: Run pre-commit hook
         run: |
-          pytest
+          uv run pre-commit run -a

.gitignore

@@ -134,3 +134,6 @@ logs/
 
 # ruff
 .ruff_cache/
+
+# mac
+.DS_Store

.pre-commit-config.yaml

@@ -12,29 +12,18 @@ repos:
     hooks:
       - id: prettier
         files: \.(css|less|scss|json|yaml|yml|html|md|mdx|js|jsx|ts|tsx|graphql|gql|vue|svelte|toml)$
-  - repo: https://github.com/astral-sh/ruff-pre-commit
-    # Ruff version.
-    rev: v0.1.3
-    hooks:
-      - id: ruff
-        args: ["--fix"]
   - repo: local
     hooks:
-      - id: black
-        name: black
-        language: system
-        entry: black
-        types: [python]
-      - id: isort
-        name: isort
+      - id: ruff
+        name: ruff
         language: system
-        entry: isort
-        args: ["--profile", "black"]
-        types: [python]
+        entry: ruff check . --fix
+        pass_filenames: false
+        always_run: true
       - id: mypy
         name: mypy
         language: system
-        entry: mypy denoisers
+        entry: mypy .
         pass_filenames: false
         always_run: true
       - id: pytest

README.md

@@ -6,13 +6,14 @@ Denoisers is a denoising library for audio with a focus on simplicity and ease o
 
 [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/wrice/denoisers)
 
-
 ## Usage/Examples
 
 ```sh
 pip install denoisers
 ```
 
+### Inference
+
 ```python
 import torch
 import torchaudio
@@ -43,3 +44,19 @@ for i in tqdm(range(0, padded.shape[-1], chunk_size)):
 
 denoised = torch.concat(clean, 1)[:, :audio.shape[-1]]
 ```
+
+### Train
+
+```sh
+
+train unet1d unet1d-24khz /data_root/
+
+```
+
+### Publish
+
+```sh
+
+publish model model_name /path/to/model
+
+```

bin/run

@@ -2,7 +2,6 @@
 
 set -e
 
-subsys=$1
-command=$2
-shift; shift
-exec python -m "denoisers.scripts.$subsys.$command" "$@"
+command=$1
+shift
+exec python -m "denoisers.scripts.$command" "$@"