Enable RUF018 rule for walrus assignments in asserts (#18886)

(cherry picked from commit 018a308)

Author: awaelchli

.pre-commit-config.yaml

@@ -83,6 +83,12 @@ repos:
           - flake8-simplify
           - flake8-return
 
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: "v0.1.3"
+    hooks:
+      - id: ruff
+        args: ["--fix", "--preview"]
+
   - repo: https://github.com/psf/black
     rev: 23.9.1
     hooks:
@@ -120,9 +126,3 @@ repos:
       - id: prettier
         # https://prettier.io/docs/en/options.html#print-width
         args: ["--print-width=120"]
-
-  - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: "v0.0.292"
-    hooks:
-      - id: ruff
-        args: ["--fix"]

pyproject.toml

@@ -53,6 +53,7 @@ select = [
     "E", "W",  # see: https://pypi.org/project/pycodestyle
     "F",  # see: https://pypi.org/project/pyflakes
     "S",  # see: https://pypi.org/project/flake8-bandit
+    "RUF018",  # see: https://docs.astral.sh/ruff/rules/assignment-in-assert
 ]
 extend-select = [
     "I",  # see: isort
@@ -64,6 +65,7 @@ extend-select = [
 ignore = [
     "E731",  # Do not assign a lambda expression, use a def
     "S108",
+    "E203", # conflicts with black
 ]
 # Exclude a variety of commonly ignored directories.
 exclude = [

src/lightning/app/cli/connect/app.py

@@ -112,7 +112,7 @@ def connect_app(app_name_or_id: str):
 
             for command_name, metadata in retriever.api_commands.items():
                 if "cls_path" in metadata:
-                    target_file = os.path.join(commands_folder, f"{command_name.replace(' ','_')}.py")
+                    target_file = os.path.join(commands_folder, f"{command_name.replace(' ', '_')}.py")
                     _download_command(
                         command_name,
                         metadata["cls_path"],

src/lightning/app/source_code/tar.py

@@ -97,7 +97,7 @@ def _get_split_size(
     max_size = max_split_count * (1 << 31)  # max size per part limited by Requests or urllib as shown in ref above
     if total_size > max_size:
         raise click.ClickException(
-            f"The size of the datastore to be uploaded is bigger than our {max_size/(1 << 40):.2f} TBytes limit"
+            f"The size of the datastore to be uploaded is bigger than our {max_size / (1 << 40):.2f} TBytes limit"
         )
 
     split_size = minimum_split_size

src/lightning/pytorch/callbacks/lr_monitor.py

@@ -272,8 +272,8 @@ def _add_prefix(
     def _add_suffix(self, name: str, param_groups: List[Dict], param_group_index: int, use_names: bool = True) -> str:
         if len(param_groups) > 1:
             if not use_names:
-                return f"{name}/pg{param_group_index+1}"
-            pg_name = param_groups[param_group_index].get("name", f"pg{param_group_index+1}")
+                return f"{name}/pg{param_group_index + 1}"
+            pg_name = param_groups[param_group_index].get("name", f"pg{param_group_index + 1}")
             return f"{name}/{pg_name}"
         if use_names:
             pg_name = param_groups[param_group_index].get("name")

src/lightning/pytorch/cli.py

@@ -627,8 +627,8 @@ def get_automatic(
         if len(optimizers) > 1 or len(lr_schedulers) > 1:
             raise MisconfigurationException(
                 f"`{self.__class__.__name__}.add_configure_optimizers_method_to_model` expects at most one optimizer "
-                f"and one lr_scheduler to be 'AUTOMATIC', but found {optimizers+lr_schedulers}. In this case the user "
-                "is expected to link the argument groups and implement `configure_optimizers`, see "
+                f"and one lr_scheduler to be 'AUTOMATIC', but found {optimizers + lr_schedulers}. In this case the "
+                "user is expected to link the argument groups and implement `configure_optimizers`, see "
                 "https://lightning.ai/docs/pytorch/stable/common/lightning_cli.html"
                 "#optimizers-and-learning-rate-schedulers"
             )

tests/tests_app/core/test_lightning_api.py

@@ -537,7 +537,7 @@ def test_configure_api():
     asyncio.set_event_loop(loop)
     results = loop.run_until_complete(asyncio.gather(*coros))
     response_time = time() - t0
-    print(f"RPS: {N/response_time}")
+    print(f"RPS: {N / response_time}")
     assert response_time < 10
     assert len(results) == N
     assert all(r.get("detail", None) == ("HERE" if i % 5 == 0 else None) for i, r in enumerate(results))

tests/tests_app/storage/test_copier.py

@@ -63,7 +63,7 @@ def test_copier_handles_exception(stat_mock, dir_mock, monkeypatch):
     copy_request_queue.put(request)
     copier.run_once()
     response = copy_response_queue.get()
-    assert type(response.exception) == OSError
+    assert type(response.exception) is OSError
     assert response.exception.args[0] == "Something went wrong"
 
 

tests/tests_app/utilities/test_cli_helpers.py

@@ -51,11 +51,11 @@ def test_arrow_time_callback():
     assert _arrow_time_callback(Mock(), Mock(), "2022-08-23 12:34:00.000") == arrow.Arrow(2022, 8, 23, 12, 34)
 
     # Just check humanized format is parsed
-    assert type(_arrow_time_callback(Mock(), Mock(), "48 hours ago")) == arrow.Arrow
+    assert type(_arrow_time_callback(Mock(), Mock(), "48 hours ago")) is arrow.Arrow
 
-    assert type(_arrow_time_callback(Mock(), Mock(), "60 minutes ago")) == arrow.Arrow
+    assert type(_arrow_time_callback(Mock(), Mock(), "60 minutes ago")) is arrow.Arrow
 
-    assert type(_arrow_time_callback(Mock(), Mock(), "120 seconds ago")) == arrow.Arrow
+    assert type(_arrow_time_callback(Mock(), Mock(), "120 seconds ago")) is arrow.Arrow
 
     # Check raising errors
     with pytest.raises(Exception, match="cannot parse time Mon"):

tests/tests_app/utilities/test_exceptions.py

@@ -59,7 +59,7 @@ def test_4xx_exceptions_caught_in_subcommands(self, mock_api_handled_group, mock
 
         mock_subcommand.invoke.assert_called
         assert result.exit_code == 1
-        assert type(result.exception) == ClickException
+        assert type(result.exception) is ClickException
         assert api_error_msg == str(result.exception)
 
     def test_original_thrown_if_cannot_decode_body(self, mock_api_handled_group, mock_subcommand):
@@ -81,4 +81,4 @@ def test_original_thrown_if_cannot_decode_body(self, mock_api_handled_group, moc
 
         mock_subcommand.invoke.assert_called
         assert result.exit_code == 1
-        assert type(result.exception) == ApiException
+        assert type(result.exception) is ApiException