Skip to content

Commit c97ae0d

Browse files
committed
fix: fix runif tensorrt logic.
1 parent 0723071 commit c97ae0d

File tree

1 file changed

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

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,7 @@ def _runif_reasons(
9797
if onnx and not _ONNX_AVAILABLE:
9898
reasons.append("onnx")
9999

100-
if onnx and not _TORCH_TRT_AVAILABLE:
100+
if tensorrt and not _TORCH_TRT_AVAILABLE:
101101
reasons.append("torch-tensorrt")
102102

103103
return reasons, kwargs

0 commit comments

Comments
 (0)