We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 0723071 commit c97ae0dCopy full SHA for c97ae0d
src/lightning/pytorch/utilities/testing/_runif.py
@@ -97,7 +97,7 @@ def _runif_reasons(
97
if onnx and not _ONNX_AVAILABLE:
98
reasons.append("onnx")
99
100
- if onnx and not _TORCH_TRT_AVAILABLE:
+ if tensorrt and not _TORCH_TRT_AVAILABLE:
101
reasons.append("torch-tensorrt")
102
103
return reasons, kwargs
0 commit comments