[pre-commit.ci] pre-commit suggestions (#226)

* [pre-commit.ci] pre-commit suggestions

updates:
- [github.com/psf/black: 23.11.0 → 24.1.1](psf/black@23.11.0...24.1.1)
- [github.com/pre-commit/mirrors-prettier: v3.1.0 → v4.0.0-alpha.8](pre-commit/mirrors-prettier@v3.1.0...v4.0.0-alpha.8)
- [github.com/astral-sh/ruff-pre-commit: v0.1.6 → v0.2.0](astral-sh/ruff-pre-commit@v0.1.6...v0.2.0)

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* Apply suggestions from code review

---------

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

.pre-commit-config.yaml

@@ -43,7 +43,7 @@ repos:
         args: ["--in-place"]
 
   - repo: https://github.com/psf/black
-    rev: 23.11.0
+    rev: 24.1.1
     hooks:
       - id: black
         name: Black code
@@ -77,7 +77,7 @@ repos:
           - flake8-bugbear
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.1.6
+    rev: v0.2.0
     hooks:
       - id: ruff
         args: ["--fix"]

src/lightning_utilities/core/__init__.py

@@ -1,4 +1,5 @@
 """Core utilities."""
+
 from lightning_utilities.core.apply_func import apply_to_collection
 from lightning_utilities.core.enums import StrEnum
 from lightning_utilities.core.imports import compare_version, module_available

src/lightning_utilities/core/rank_zero.py

@@ -18,13 +18,11 @@
 
 
 @overload
-def rank_zero_only(fn: Callable[P, T]) -> Callable[P, Optional[T]]:
-    ...
+def rank_zero_only(fn: Callable[P, T]) -> Callable[P, Optional[T]]: ...
 
 
 @overload
-def rank_zero_only(fn: Callable[P, T], default: T) -> Callable[P, T]:
-    ...
+def rank_zero_only(fn: Callable[P, T], default: T) -> Callable[P, T]: ...
 
 
 def rank_zero_only(fn: Callable[P, T], default: Optional[T] = None) -> Callable[P, Optional[T]]:

tests/unittests/core/test_inheritance.py

@@ -2,20 +2,15 @@
 
 
 def test_get_all_subclasses():
-    class A1:
-        ...
+    class A1: ...
 
-    class A2(A1):
-        ...
+    class A2(A1): ...
 
-    class B1:
-        ...
+    class B1: ...
 
-    class B2(B1):
-        ...
+    class B2(B1): ...
 
-    class C(A2, B2):
-        ...
+    class C(A2, B2): ...
 
     assert get_all_subclasses(A1) == {A2, C}
     assert get_all_subclasses(A2) == {C}

tests/unittests/core/test_overrides.py

@@ -8,23 +8,19 @@
 
 
 class LightningModule:
-    def training_step(self):
-        ...
+    def training_step(self): ...
 
 
 class BoringModel(LightningModule):
-    def training_step(self):
-        ...
+    def training_step(self): ...
 
 
 class Strategy:
     @contextmanager
-    def model_sharded_context():
-        ...
+    def model_sharded_context(): ...
 
 
-class SingleDeviceStrategy(Strategy):
-    ...
+class SingleDeviceStrategy(Strategy): ...
 
 
 def test_is_overridden():

tests/unittests/test/test_warnings.py

@@ -15,8 +15,7 @@ def test_no_warning_call():
     with no_warning_call(DeprecationWarning):
         warnings.warn("foo")
 
-    class MyDeprecationWarning(DeprecationWarning):
-        ...
+    class MyDeprecationWarning(DeprecationWarning): ...
 
     with pytest.raises(
         AssertionError, match=escape("`DeprecationWarning` was raised: MyDeprecationWarning('bar')")