Skip to content

Commit

Permalink
Lint fix
Browse files Browse the repository at this point in the history
  • Loading branch information
dagardner-nv committed Jan 17, 2025
1 parent 5f64f05 commit f541415
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions python/morpheus/morpheus/utils/onnx_to_trt.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,11 +21,11 @@
try:
import tensorrt as trt
except ImportError as e:
error_msg = ("The ONNX to TensorRT conversion utility requires additional packages, which can be installed using "
"the following command:\n"
"conda env update --solver=libmamba -n morpheus --file "
"conda/environments/model-utils_cuda-125_arch-$(arch).yaml")
raise ImportError(error_msg) from e
raise ImportError(
"The ONNX to TensorRT conversion utility requires additional packages, which can be installed using the "
"following command:\n"
"conda env update --solver=libmamba -n morpheus --file "
"conda/environments/model-utils_cuda-125_arch-$(arch).yaml") from e


def gen_engine(config: ConfigOnnxToTRT):
Expand Down

0 comments on commit f541415

Please sign in to comment.