Refactor with black

pydantic · Feb 18, 2024 · 2813563 · 2813563
1 parent a317c97
commit 2813563
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 13 deletions.
diff --git a/bump_pydantic/codemods/class_def_visitor.py b/bump_pydantic/codemods/class_def_visitor.py
@@ -9,6 +9,7 @@
 1.1. If it is, and if any `base_model_cls` is found, remove from `cls`, and add to `base_model_cls`.
 1.2. If it's not, it continues on the `cls`
 """
+
 from __future__ import annotations
 
 from collections import defaultdict

diff --git a/bump_pydantic/codemods/replace_config.py b/bump_pydantic/codemods/replace_config.py
@@ -215,20 +215,22 @@ def leave_config_class_childless(self, original_node: cst.ClassDef, updated_node
         AddImportsVisitor.add_needed_import(context=self.context, **needed_import)  # type: ignore[arg-type]
         block = cst.ensure_type(updated_node.body, cst.IndentedBlock)
         body = [
-            cst.SimpleStatementLine(
-                body=[
-                    cst.Assign(
-                        targets=[cst.AssignTarget(target=cst.Name("model_config"))],
-                        value=cst.Call(
-                            func=cst.Name("SettingsConfigDict" if self.is_base_settings else "ConfigDict"),
-                            args=self.config_args,
-                        ),
-                    )
-                ],
-                leading_lines=self._leading_lines_from_removed_keys(self.config_args),
+            (
+                cst.SimpleStatementLine(
+                    body=[
+                        cst.Assign(
+                            targets=[cst.AssignTarget(target=cst.Name("model_config"))],
+                            value=cst.Call(
+                                func=cst.Name("SettingsConfigDict" if self.is_base_settings else "ConfigDict"),
+                                args=self.config_args,
+                            ),
+                        )
+                    ],
+                    leading_lines=self._leading_lines_from_removed_keys(self.config_args),
+                )
+                if m.matches(statement, m.ClassDef(name=m.Name(value="Config")))
+                else statement
             )
-            if m.matches(statement, m.ClassDef(name=m.Name(value="Config")))
-            else statement
             for statement in block.body
         ]
         self.is_base_settings = False

diff --git a/bump_pydantic/codemods/replace_imports.py b/bump_pydantic/codemods/replace_imports.py
@@ -7,6 +7,7 @@
 4. `from pydantic.settings import BaseSettings as <name>`  # TODO: This is not working.
 5. `import pydantic` -> `pydantic.BaseSettings`
 """
+
 from __future__ import annotations
 
 from dataclasses import dataclass