Skip to content

Commit 02757a3

Browse files
committed
docs: add tensorrt description in docstring.
1 parent bf63faa commit 02757a3

File tree

1 file changed

+1
-0
lines changed
  • src/lightning/pytorch/utilities/testing

1 file changed

+1
-0
lines changed

src/lightning/pytorch/utilities/testing/_runif.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ def _runif_reasons(
6565
psutil: Require that psutil is installed.
6666
sklearn: Require that scikit-learn is installed.
6767
onnx: Require that onnx is installed.
68+
tensorrt: Require that torch-tensorrt is installed.
6869
6970
"""
7071

0 commit comments

Comments
 (0)