Apply linter

Author: daniel-gallo

.github/workflows/python.yml

@@ -26,7 +26,7 @@ jobs:
     - run: pip install -r requirements_github.txt
     - name: Lint with Ruff
       run: |
-        ruff check --select I --output-format=github .
+        ruff check --output-format=github .
     - name: Test with pytest
       run: |
         python -m pytest tests

dataset.py

@@ -4,7 +4,6 @@
 import jax
 import tensorflow as tf
 
-
 AUTOTUNE = tf.data.AUTOTUNE
 
 

pyproject.toml

@@ -0,0 +1,9 @@
+[project]
+name = "naraim"
+requires-python = "==3.11"
+
+[tool.ruff]
+line-length = 120
+
+[tool.ruff.lint]
+select = ["E", "F", "I"]

trainer.py

@@ -16,8 +16,8 @@
 from torch.utils.tensorboard import SummaryWriter
 from tqdm import tqdm, trange
 
-from exponential_scheduler import warmup_exponential_decay_cooldown_scheduler
 from dataset import prefetch
+from exponential_scheduler import warmup_exponential_decay_cooldown_scheduler
 from model import ClassificationModel, PretrainingModel
 from model.classification import NoTransformerClassificationModel