apply pre-commit

.github/workflows/ci.yml

@@ -18,7 +18,7 @@ jobs:
     name: Test Suite
     runs-on: ubuntu-latest
     steps:
-      - 
+      -
         name: Checkout the repository
         uses: actions/checkout@v3
       -
@@ -27,7 +27,7 @@ jobs:
         with:
           registry: ${{ env.REGISTRY }}
           username: ${{ github.actor }}
-          password: ${{ secrets.GITHUB_TOKEN }} 
+          password: ${{ secrets.GITHUB_TOKEN }}
       -
         name: Compile and Run the tests
         run: ./run_tests.sh
@@ -36,7 +36,7 @@ jobs:
     name: Rustfmt
     runs-on: ubuntu-latest
     steps:
-      - 
+      -
         name: Checkout the repository
         uses: actions/checkout@v3
       -
@@ -49,12 +49,12 @@ jobs:
       -
         name: Compile and Run the tests
         run: ./run_fmt.sh
-        
+
   clippy:
     name: Clippy
     runs-on: ubuntu-latest
     steps:
-      - 
+      -
         name: Checkout the repository
         uses: actions/checkout@v3
       -

.github/workflows/devel_docker.yml

@@ -26,7 +26,7 @@ jobs:
         with:
           registry: ${{ env.REGISTRY }}
           username: ${{ github.actor }}
-          password: ${{ secrets.GITHUB_TOKEN }} 
+          password: ${{ secrets.GITHUB_TOKEN }}
       -
         name: Build and push Docker image
         uses: docker/build-push-action@v2

.github/workflows/release.yml

@@ -21,7 +21,7 @@ jobs:
         with:
           registry: ${{ env.REGISTRY }}
           username: ${{ github.actor }}
-          password: ${{ secrets.GITHUB_TOKEN }} 
+          password: ${{ secrets.GITHUB_TOKEN }}
       - uses: messense/maturin-action@v1
         with:
           container: ghcr.io/kornia/kornia-rs/release:latest
@@ -68,7 +68,7 @@ jobs:
       matrix:
         python-version: ["3.7", "3.8", "3.9", "3.10", "3.11"]
         target: [x64, x86]
-    steps: 
+    steps:
       - uses: ilammy/setup-nasm@v1
       - uses: actions/checkout@v3
       - uses: actions/setup-python@v2

.github/workflows/release_docker.yml

@@ -26,7 +26,7 @@ jobs:
         with:
           registry: ${{ env.REGISTRY }}
           username: ${{ github.actor }}
-          password: ${{ secrets.GITHUB_TOKEN }} 
+          password: ${{ secrets.GITHUB_TOKEN }}
       -
         name: Build and push Docker image
         uses: docker/build-push-action@v2

.gitignore

@@ -69,4 +69,4 @@ docs/_build/
 .vscode/
 
 # Pyenv
-.python-version
+.python-version

.pre-commit-config.yaml

@@ -0,0 +1,16 @@
+repos:
+-   repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v3.1.0
+    hooks:
+    -   id: check-byte-order-marker
+    -   id: check-case-conflict
+    -   id: check-merge-conflict
+    -   id: check-symlinks
+    -   id: check-yaml
+    -   id: end-of-file-fixer
+    -   id: mixed-line-ending
+    -   id: trailing-whitespace
+-   repo: https://github.com/pre-commit/pre-commit
+    rev: v2.5.1
+    hooks:
+    -   id: validate_manifest

.pre-commit-hooks.yaml

@@ -0,0 +1,22 @@
+- id: fmt
+  name: fmt
+  description: Format files with cargo fmt.
+  entry: cargo fmt
+  language: system
+  types: [rust]
+  args: ["--"]
+- id: cargo-check
+  name: cargo check
+  description: Check the package for errors.
+  entry: cargo check
+  language: system
+  types: [rust]
+  pass_filenames: false
+- id: clippy
+  name: clippy
+  description: Lint rust sources
+  entry: cargo clippy
+  language: system
+  args: ["--", "-D", "warnings"]
+  types: [rust]
+  pass_filenames: false

Cargo.toml

@@ -25,4 +25,4 @@ harness = false
 
 [features]
 default = []
-viz = ["dep:vviz"]
+viz = ["dep:vviz"]

devel.sh

@@ -1,3 +1,3 @@
 #!/bin/bash -e
 DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
-$DIR/docker/devel.sh $@
+$DIR/docker/devel.sh $@

docker/devel.Dockerfile

@@ -23,5 +23,10 @@ RUN apt-get update --fix-missing && \
     apt-get clean
 
 RUN pip3 install maturin[patchelf]
+RUN pip3 install pre-commit
+
+# add rust tools
+RUN rustup component add rustfmt
+RUN rustup component add clippy
 
 WORKDIR /workspace

kornia-rs.code-workspace

@@ -4,4 +4,4 @@
 			"path": "."
 		}
 	]
-}
+}

pyproject.toml

@@ -35,4 +35,4 @@ dev = [
 [project.urls]
 homepage = "https://www.kornia.org"
 documentation = "https://kornia.readthedocs.io"
-repository = "https://github.com/kornia/kornia-rs"
+repository = "https://github.com/kornia/kornia-rs"

release.sh

@@ -1,3 +1,3 @@
 #!/bin/bash -e
 DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
-$DIR/docker/release.sh $@
+$DIR/docker/release.sh $@

run_clippy.sh

@@ -1,2 +1,2 @@
 #!/bin/bash -e
-./devel.sh "rustup component add clippy; cargo clippy -- -D warnings"
+./devel.sh "rustup component add clippy; cargo clippy -- -D warnings"

run_fmt.sh

@@ -1,2 +1,2 @@
 #!/bin/bash -e
-./devel.sh "rustup component add rustfmt; cargo fmt --all -- --check"
+./devel.sh "rustup component add rustfmt; cargo fmt --all -- --check"

run_tests.sh

@@ -1,2 +1,2 @@
 #!/bin/bash -e
-./devel.sh "python3 -m venv .venv; source .venv/bin/activate; maturin develop --extras dev --all-features; pytest test/"
+./devel.sh "python3 -m venv .venv; source .venv/bin/activate; maturin develop --extras dev --all-features; pytest test/"

test/test_io.py

@@ -50,4 +50,4 @@ def test_decompress():
     assert image_size.height == 195
 
     decoded_tensor: K.cvTensor = image_decoder.decode(bytes(image_encoded))
-    decoded_image = np.from_dlpack(decoded_tensor)
+    decoded_image = np.from_dlpack(decoded_tensor)

test/test_tensor.py

@@ -30,4 +30,4 @@ def test_conversions2():
     # to dlpack / torch / numpy
     th_tensor = torch.utils.dlpack.from_dlpack(cv_tensor)
     np_array = np.from_dlpack(cv_tensor)
-    np.testing.assert_array_equal(np_array, th_tensor.numpy())
+    np.testing.assert_array_equal(np_array, th_tensor.numpy())