chore: squash commit with main

.github/workflows/docs.yaml

@@ -13,7 +13,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: 3.8.15
+          python-version: 3.10.8
       - name: Install and configure Poetry
         uses: snok/install-poetry@v1
         with:

.github/workflows/pytest.yaml

@@ -1,5 +1,5 @@
 name: tests
-on: [pull_request]
+on: [push]
 jobs:
   test:
     runs-on: ubuntu-latest
@@ -10,7 +10,7 @@ jobs:
       - name: Set up Python
         uses: actions/setup-python@v2
         with:
-          python-version: 3.8.15
+          python-version: 3.10.8
       - name: Install and configure Poetry
         uses: snok/install-poetry@v1
         with:
@@ -25,11 +25,11 @@ jobs:
           key: venv-${{ runner.os }}-${{ steps.setup-python.outputs.python-version }}-${{ hashFiles('**/poetry.lock') }}
       - name: Install dependencies
         if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true'
-        run: poetry install --without docs --no-interaction --no-root
+        run: poetry install --no-interaction --no-root
       - name: Install library
-        run: poetry install --no-interaction --without docs
+        run: poetry install --no-interaction
       - name: Run tests
-        run: poetry run pytest --doctest-modules --cov=src/ --cov-report=xml
+        run: poetry run pytest
       - name: Upload coverage to Codecov
         uses: codecov/codecov-action@v3
         with:

.gitignore

@@ -10,3 +10,5 @@ docs/assets/schemas/
 mock_data/
 notebooks/wandb/
 src/wandb/
+src/airflow/logs/
+site/

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.4.0
+    rev: v4.5.0
     hooks:
       - id: trailing-whitespace
       - id: end-of-file-fixer
@@ -26,34 +26,17 @@ repos:
       - id: python-use-type-annotations
       - id: python-check-blanket-noqa
 
-  - repo: https://github.com/asottile/yesqa
-    rev: v1.4.0
-    hooks:
-      - id: yesqa
-        additional_dependencies: &flake8_deps
-          - flake8-annotations==2.9.0
-          - flake8-broken-line==0.5.0
-          - flake8-bugbear==22.7.1
-          - flake8-comprehensions==3.10.0
-          - flake8-eradicate==1.3.0
-          - flake8-quotes==3.3.1
-          - flake8-simplify==0.19.3
-          - flake8-tidy-imports==4.8.0
-          - flake8-type-checking==2.1.2
-          - flake8-typing-imports==1.12.0
-          - flake8-use-fstring==1.4
-          - flake8-class-attributes-order==0.1.3
-          - flake8-docstrings
-          - flake8-pytest-style
-          - pep8-naming==0.13.1
-          - darglint
-
   - repo: https://github.com/hadialqattan/pycln
-    rev: v2.1.3
+    rev: v2.2.2
     hooks:
       - id: pycln
         args: [--all]
 
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.0.292
+    hooks:
+      - id: ruff
+
   - repo: https://github.com/pycqa/isort
     rev: 5.12.0
     hooks:
@@ -62,25 +45,19 @@ repos:
 
 
   - repo: https://github.com/psf/black
-    rev: 23.1.0
+    rev: 23.9.1
     hooks:
       - id: black
 
-  - repo: https://github.com/pycqa/flake8
-    rev: 5.0.4
-    hooks:
-      - id: flake8
-        additional_dependencies: *flake8_deps
-
   - repo: https://github.com/alessandrojcm/commitlint-pre-commit-hook
-    rev: v9.4.0
+    rev: v9.5.0
     hooks:
       - id: commitlint
         additional_dependencies: ['@commitlint/config-conventional']
         stages: [commit-msg]
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: 'v0.991'
+    rev: 'v1.5.1'
     hooks:
       - id: mypy
 

.python-version

@@ -1 +1 @@
-3.8.15
+3.10.8

.vscode/extensions.json

@@ -0,0 +1,12 @@
+{
+  "recommendations": [
+    "charliermarsh.ruff",
+    "ms-python.isort",
+    "ms-python.mypy-type-checker",
+    "ms-python.python",
+    "ms-python.black-formatter"
+  ],
+  "unwantedRecommendations": [
+    "ms-python.flake8"
+  ]
+}

.vscode/settings.json

@@ -1,7 +1,12 @@
 {
   "[python]": {
+    "editor.defaultFormatter": "ms-python.black-formatter",
     "editor.formatOnPaste": false,
-    "editor.formatOnSave": true
+    "editor.formatOnSave": true,
+    "editor.codeActionsOnSave": {
+      "source.fixAll": false,
+      "source.organizeImports": true
+    }
   },
   "python.terminal.launchArgs": [
     "-m",
@@ -18,18 +23,6 @@
     }
   },
   "json.format.keepLines": true,
-  "python.linting.flake8Enabled": true,
-  "python.linting.flake8CategorySeverity.F": "Information",
-  "python.linting.flake8CategorySeverity.E": "Information",
-  "python.linting.flake8CategorySeverity.W": "Information",
-  "python.formatting.blackArgs": [],
-  "python.linting.mypyEnabled": true,
-  "python.linting.mypyCategorySeverity.error": "Information",
-  "python.linting.pydocstyleEnabled": true,
-  "python.linting.pydocstyleArgs": [
-    "--convention=google"
-  ],
-  "python.formatting.provider": "black",
   "isort.args": [
     "--profile",
     "black"
@@ -42,5 +35,9 @@
   ],
   "python.testing.unittestEnabled": false,
   "python.testing.pytestEnabled": true,
-  "docwriter.style": "Google"
+  "mypy-type-checker.severity": {
+    "error": "Information"
+  },
+  "ruff.fixAll": false,
+  "ruff.organizeImports": false
 }