Add config files

diff --git a/reference_transformer/.pre-commit-config.yaml b/reference_transformer/.pre-commit-config.yaml
new file mode 100644 (file)
index 0000000..eabc05c
--- /dev/null
+++ b/reference_transformer/.pre-commit-config.yaml
@@ -0,0 +1,6 @@
+repos:
+-   repo: https://github.com/psf/black
+    rev: 24.4.2
+    hooks:
+    - id: black
+      language_version: python3.12

diff --git a/reference_transformer/pyproject.toml b/reference_transformer/pyproject.toml
index 288577b334d7017917e141f246bac2a2d3943588..7acc369f23f9024b2ee2679bfb5e7d2152fc12ef 100644 (file)
--- a/reference_transformer/pyproject.toml
+++ b/reference_transformer/pyproject.toml
@@ -9,6 +9,8 @@ readme = "README.md"
 python = "^3.12"
 torch = "^2.3.1"
 numpy = "^2.0.0"
+pytest = "^8.2.2"
+pre-commit = "^3.7.1"
 
 
 [build-system]