lint: switch Black to ruff-format (#230)

* lint: switch Black to `ruff-format`

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

.pre-commit-config.yaml

@@ -42,12 +42,6 @@ repos:
         additional_dependencies: [tomli]
         args: ["--in-place"]
 
-  - repo: https://github.com/psf/black
-    rev: 24.1.1
-    hooks:
-      - id: black
-        name: Black code
-
   - repo: https://github.com/executablebooks/mdformat
     rev: 0.7.17
     hooks:
@@ -79,5 +73,7 @@ repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
     rev: v0.2.0
     hooks:
+      - id: ruff-format
+        args: ["--preview"]
       - id: ruff
         args: ["--fix"]

pyproject.toml

@@ -62,12 +62,6 @@ wrap-descriptions = 120
 blank = true
 
 
-[tool.black]
-# https://github.com/psf/black
-line-length = 120
-exclude = "(.eggs|.git|.hg|.mypy_cache|.venv|_build|buck-out|build|dist)"
-
-
 [tool.mypy]
 files = [
     "src/lightning_utilities",

scripts/adjust-torch-versions.py

@@ -2,6 +2,7 @@
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
 """Adjusting version across PTorch ecosystem."""
+
 import logging
 import os
 import re

src/lightning_utilities/cli/__main__.py

@@ -9,14 +9,12 @@
 
 def main() -> None:
     """CLI entry point."""
-    fire.Fire(
-        {
-            "requirements": {
-                "prune-pkgs": prune_pkgs_in_requirements,
-                "set-oldest": replace_oldest_ver,
-            }
+    fire.Fire({
+        "requirements": {
+            "prune-pkgs": prune_pkgs_in_requirements,
+            "set-oldest": replace_oldest_ver,
         }
-    )
+    })
 
 
 if __name__ == "__main__":

src/lightning_utilities/core/apply_func.py

@@ -221,12 +221,10 @@ def apply_to_collections(
     if isinstance(data1, Mapping) and data2 is not None:
         # use union because we want to fail if a key does not exist in both
         zipped = {k: (data1[k], data2[k]) for k in data1.keys() | data2.keys()}
-        return elem_type(
-            {
-                k: apply_to_collections(*v, dtype, function, *args, wrong_dtype=wrong_dtype, **kwargs)
-                for k, v in zipped.items()
-            }
-        )
+        return elem_type({
+            k: apply_to_collections(*v, dtype, function, *args, wrong_dtype=wrong_dtype, **kwargs)
+            for k, v in zipped.items()
+        })
 
     is_namedtuple_ = is_namedtuple(data1)
     is_sequence = isinstance(data1, Sequence) and not isinstance(data1, str)

src/lightning_utilities/core/rank_zero.py

@@ -3,6 +3,7 @@
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
 """Utilities that can be used for calling functions on a particular rank."""
+
 import logging
 import warnings
 from functools import wraps