Skip to content

Commit

Permalink
remove black changes
Browse files Browse the repository at this point in the history
  • Loading branch information
Kludex committed Mar 29, 2024
1 parent e0711e1 commit b2ea38c
Showing 1 changed file with 13 additions and 15 deletions.
28 changes: 13 additions & 15 deletions bump_pydantic/codemods/replace_config.py
Original file line number Diff line number Diff line change
Expand Up @@ -217,22 +217,20 @@ 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),
)
if m.matches(statement, m.ClassDef(name=m.Name(value="Config")))
else statement
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
for statement in block.body
]
self.is_base_settings = False
Expand Down

0 comments on commit b2ea38c

Please sign in to comment.