Skip to content

Commit e0711e1

Browse files
denisartKludex
authored andcommitted
Refactor with black
1 parent 20d6a88 commit e0711e1

File tree

1 file changed

+15
-13
lines changed

1 file changed

+15
-13
lines changed

bump_pydantic/codemods/replace_config.py

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -217,20 +217,22 @@ def leave_config_class_childless(self, original_node: cst.ClassDef, updated_node
217217
AddImportsVisitor.add_needed_import(context=self.context, **needed_import) # type: ignore[arg-type]
218218
block = cst.ensure_type(updated_node.body, cst.IndentedBlock)
219219
body = [
220-
cst.SimpleStatementLine(
221-
body=[
222-
cst.Assign(
223-
targets=[cst.AssignTarget(target=cst.Name("model_config"))],
224-
value=cst.Call(
225-
func=cst.Name("SettingsConfigDict" if self.is_base_settings else "ConfigDict"),
226-
args=self.config_args,
227-
),
228-
)
229-
],
230-
leading_lines=self._leading_lines_from_removed_keys(self.config_args),
220+
(
221+
cst.SimpleStatementLine(
222+
body=[
223+
cst.Assign(
224+
targets=[cst.AssignTarget(target=cst.Name("model_config"))],
225+
value=cst.Call(
226+
func=cst.Name("SettingsConfigDict" if self.is_base_settings else "ConfigDict"),
227+
args=self.config_args,
228+
),
229+
)
230+
],
231+
leading_lines=self._leading_lines_from_removed_keys(self.config_args),
232+
)
233+
if m.matches(statement, m.ClassDef(name=m.Name(value="Config")))
234+
else statement
231235
)
232-
if m.matches(statement, m.ClassDef(name=m.Name(value="Config")))
233-
else statement
234236
for statement in block.body
235237
]
236238
self.is_base_settings = False

0 commit comments

Comments
 (0)