From: Sondre Wold <sondrewold1@gmail.com>
Date: Mon, 17 Jun 2024 11:43:34 +0000 (+0200)
Subject: Add config files
X-Git-Url: https://letsjmore.com/?a=commitdiff_plain;h=6bd163ebf8f59139d66bcd45a7e30e71f02e441f;p=reference_transformer.git

Add config files
---

diff --git a/reference_transformer/.pre-commit-config.yaml b/reference_transformer/.pre-commit-config.yaml
new file mode 100644
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 288577b..7acc369 100644
--- 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]