Add pre-commit

SamuraiBarbi · Mar 16, 2023 · e24155f · e24155f
1 parent a5b4425
commit e24155f
Show file tree

Hide file tree

Showing 5 changed files with 130 additions and 102 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -0,0 +1,20 @@
+# See https://pre-commit.com for more information
+# See https://pre-commit.com/hooks.html for more hooks
+repos:
+-   repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v3.2.0
+    hooks:
+    -   id: trailing-whitespace
+    -   id: end-of-file-fixer
+    -   id: check-yaml
+    -   id: check-added-large-files
+-   repo: https://github.com/pycqa/isort
+    rev: 5.12.0
+    hooks:
+    -   id: isort
+        stages: [push]
+-   repo: https://github.com/psf/black
+    rev: 22.10.0
+    hooks:
+    -   id: black
+        stages: [push]
diff --git a/Makefile b/Makefile
diff --git a/poetry.lock b/poetry.lock
diff --git a/preprocess/preprocess_project.py b/preprocess/preprocess_project.py
@@ -1,12 +1,11 @@
-import os
 import glob
 import json
-
-from datasets import Dataset
-from transformers import HfArgumentParser
+import os
 
 import metrics
 from args import PreprocessProjectArgs
+from datasets import Dataset
+from transformers import HfArgumentParser
 
 
 def parse_args():

diff --git a/pyproject.toml b/pyproject.toml
@@ -13,7 +13,7 @@ transformers = "^4.27.1"
 
 
 [tool.poetry.group.dev.dependencies]
-black = "^23.1.0"
+pre-commit = "^3.1.1"
 
 [build-system]
 requires = ["poetry-core"]