AlphaFold losses

Genentech · Apr 29, 2024 · 5536dce · 5536dce
1 parent 3e4ffd0
commit 5536dce
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 2 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -0,0 +1,13 @@
+repos:
+  - hooks:
+      - id: "check-toml"
+      - id: "check-yaml"
+    repo: "https://github.com/pre-commit/pre-commit-hooks"
+    rev: "v4.6.0"
+  - hooks:
+      - args:
+        - "--fix"
+        id: "ruff"
+      - id: "ruff-format"
+    repo: "https://github.com/astral-sh/ruff-pre-commit"
+    rev: "v0.3.7"
diff --git a/pyproject.toml b/pyproject.toml
@@ -14,8 +14,19 @@ name = "beignet"
 readme = "README.md"
 requires-python = ">=3.10"
 
-[tool.ruff.format]
-docstring-code-format = true
+[tool.ruff.lint]
+select = [
+    "B",  # FLAKE8-BUGBEAR
+    "E",  # PYCODESTYLE ERRORS
+    "F",  # PYFLAKES
+    "I",  # ISORT
+    "W",  # PYCODESTYLE WARNINGS
+]
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = [
+    "F401", # MODULE IMPORTED BUT UNUSED
+]
 
 [tool.setuptools_scm]
 local_scheme = "no-local-version"
diff --git a/src/beignet/nn/__init__.py b/src/beignet/nn/__init__.py