More linting fix

This commit is contained in:
Rémi Bédard-Couture 2024-05-18 15:41:39 -04:00
parent 1879951bc0
commit be84fe8a5e

View File

@ -6,7 +6,7 @@ import numpy as np
try: try:
import tensorrt as trt import tensorrt as trt
from cuda import cuda from cuda import cuda
TRT_VERSION = int(trt.__version__[0:trt.__version__.find(".")]) TRT_VERSION = int(trt.__version__[0 : trt.__version__.find(".")])
TRT_SUPPORT = True TRT_SUPPORT = True
except ModuleNotFoundError: except ModuleNotFoundError: