ruff setup and pre-commit (pytorch#57)

.pre-commit-config.yaml

@@ -8,22 +8,9 @@ repos:
     -   id: debug-statements
     -   id: end-of-file-fixer
     -   id: trailing-whitespace
-# - repo: https://github.com/PyCQA/isort
-#   rev: 5.13.2
-#   hooks:
-#     - id: isort
-#       args: [--profile, black]
-- repo: https://github.com/PyCQA/flake8
-  rev: 7.1.1
+- repo: https://github.com/astral-sh/ruff-pre-commit
+  rev: v0.7.4
   hooks:
-   - id: flake8
-     args: ['--ignore=E501,E203,E731,W503']
-- repo: https://github.com/psf/black
-  rev: 24.10.0
-  hooks:
-    - id: black
-# - repo: https://github.com/pre-commit/mirrors-mypy
-#   rev: v1.10.0
-#   hooks:
-#     - id: mypy
-#       additional_dependencies: [types-dataclasses, numpy]
+    - id: ruff
+      args: [ --fix ]
+    - id: ruff-format

pyproject.toml

@@ -32,20 +32,6 @@ dev-dependencies = [
 [tool.setuptools]
 packages = ["low_bits_training", "low_bits_training.datasets", "low_bits_training.models"]
 
-[tool.black]
+[tool.ruff]
 line-length = 90
-target-version = ['py310']
-
-[tool.isort]
-line_length = 90
-py_version = 310
-
-[tool.mypy]
-python_version = "3.10"
-show_error_codes = true
-warn_redundant_casts = true
-warn_unused_ignores = true
-warn_unused_configs = true
-check_untyped_defs = true
-disallow_any_generics = true
-no_implicit_optional = false
+target-version = 'py310'

scripts/download_dataset.py

@@ -47,7 +47,6 @@ def sync_between_nodes(source_path, source_host, dest_path, dest_host):
 
 
 def main(args):
-
     run = wandb.init(
         entity=args.entity,
         project=args.project,

scripts/titan_sweep.py

@@ -17,6 +17,7 @@
  python sweep.py --config '{"command":"./train.sh", ...}' # JSON string
 
 """
+
 import json
 import argparse
 from itertools import product