diff --git a/ethosu/vela/tensor_allocation.py b/ethosu/vela/tensor_allocation.py
index c8b5129..ab65740 100644
--- a/ethosu/vela/tensor_allocation.py
+++ b/ethosu/vela/tensor_allocation.py
@@ -128,7 +128,12 @@
     print("\n" + "#" * 80)
     sg_placement = (
         sg.placement.name
-        if mem_type_set.intersection((MemType.Permanent_NPU, MemType.Permanent_CPU,))
+        if mem_type_set.intersection(
+            (
+                MemType.Permanent_NPU,
+                MemType.Permanent_CPU,
+            )
+        )
         else "Cpu and Npu"
     )
     print(
@@ -141,7 +146,15 @@
     min_mem_usage_for_alloc = max(memory_hist)
     print("Start Time -   End Time: Start Addr -   End Addr: Tensor Size: Memory Usage:  Tensor Purpose: Tensor Name")
     for start_time, end_time, size, start_addr, end_addr, purpose, name in sorted(
-        (lr.start_time, lr.end_time, lr.size, tens.address, tens.address + lr.size, tens.purpose, tens.name,)
+        (
+            lr.start_time,
+            lr.end_time,
+            lr.size,
+            tens.address,
+            tens.address + lr.size,
+            tens.purpose,
+            tens.name,
+        )
         for tens, lr in lrs.ranges.items()
     ):
         print(
@@ -184,7 +197,11 @@
 ):
     # Allocates addresses to tensors, returns False if tensors could not be fit within max_size
     lrs = live_range.extract_live_ranges_from_cascaded_passes(
-        sg, mem_area, mem_type_set, lr_graph=lr_graph, cpu_tensor_alignment=cpu_tensor_alignment,
+        sg,
+        mem_area,
+        mem_type_set,
+        lr_graph=lr_graph,
+        cpu_tensor_alignment=cpu_tensor_alignment,
     )
     total_sz = 0
     if lrs.ranges: