More linting

This commit is contained in:
Rémi Bédard-Couture 2024-05-18 16:05:06 -04:00
parent be84fe8a5e
commit 4695f8284a

View File

@ -109,7 +109,9 @@ class TensorRtDetector(DetectionApi):
def _execute(self): def _execute(self):
if TRT_VERSION < 10: if TRT_VERSION < 10:
return self.context.execute_async_v2(bindings=self.bindings, stream_handle=self.stream) return self.context.execute_async_v2(
bindings=self.bindings, stream_handle=self.stream
)
else: else:
return self.context.execute_v2(self.bindings) return self.context.execute_v2(self.bindings)