diff --git a/python/ark/runtime.py b/python/ark/runtime.py index f3baf399..1490cdeb 100644 --- a/python/ark/runtime.py +++ b/python/ark/runtime.py @@ -90,7 +90,10 @@ def launch( # Recompile if the previous launch was not compiled with the same info # or if this is the first launch - if plan_str != self.executor.plan() or device_id != self.executor.device_id(): + if ( + plan_str != self.executor.plan() + or device_id != self.executor.device_id() + ): self.executor.compile(plan_str, device_id) self.executor.launch(stream, loop_mode)