diff --git a/src/engines/jetson_tensorrt_engine.pyx b/src/engines/jetson_tensorrt_engine.pyx index cc9fc56..0826f32 100644 --- a/src/engines/jetson_tensorrt_engine.pyx +++ b/src/engines/jetson_tensorrt_engine.pyx @@ -1,5 +1,6 @@ import os import tempfile +cimport constants_inf from engines.tensorrt_engine cimport TensorRTEngine from loader_http_client cimport LoaderHttpClient, LoadResult @@ -21,7 +22,6 @@ cdef class JetsonTensorRTEngine(TensorRTEngine): @staticmethod def _download_calib_cache(LoaderHttpClient loader_client, str models_dir): cdef LoadResult res - import constants_inf try: res = loader_client.load_big_small_resource( constants_inf.INT8_CALIB_CACHE_FILE, models_dir