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

for more information, see https://pre-commit.ci
TylerYep · Mar 18, 2024 · 047f8a8 · 047f8a8
1 parent 4440a22
commit 047f8a8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/torchinfo/layer_info.py b/torchinfo/layer_info.py
@@ -217,9 +217,9 @@ def calculate_num_params(self) -> None:
                 final_name = name
         # Fix the final row to display more nicely
         if self.inner_layers:
-            self.inner_layers[final_name][
-                ColumnSettings.NUM_PARAMS
-            ] = f"└─{self.inner_layers[final_name][ColumnSettings.NUM_PARAMS][2:]}"
+            self.inner_layers[final_name][ColumnSettings.NUM_PARAMS] = (
+                f"└─{self.inner_layers[final_name][ColumnSettings.NUM_PARAMS][2:]}"
+            )
 
     def calculate_macs(self) -> None:
         """