Refactor ocrs_models.datasets

.github/workflows/ci.yml

@@ -25,7 +25,6 @@ jobs:
         run: |
           poetry install
           poetry run pip install torch torchvision
-      - name: Check formatting and types
+      - name: Lint, format and typecheck
         run: |
-          poetry run black --check .
-          poetry run mypy ocrs_models
+          make qa

.gitignore

@@ -1,5 +1,7 @@
 # Model checkpoints
 *.pt
 
-datasets/
-wandb/
+__pycache__/
+
+/datasets
+/wandb

Makefile

@@ -1,3 +1,18 @@
-qa:
+.PHONY: qa
+qa: checkformat lint typecheck
+
+.PHONY: checkformat
+checkformat:
 	poetry run black --check .
+
+.PHONY: format
+format:
+	poetry run black ocrs_models
+
+.PHONY: lint
+lint:
+	poetry run ruff ocrs_models
+
+.PHONY: typecheck
+typecheck:
 	poetry run mypy ocrs_models