Skip to content

Commit 6422746

Browse files
authored
ci: Remove trt-llm breakpoint (#13499)
* tests: Disable flaky test Signed-off-by: oliver könig <[email protected]> * remove breakpoint Signed-off-by: oliver könig <[email protected]> --------- Signed-off-by: oliver könig <[email protected]>
1 parent 39e6bd7 commit 6422746

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

reinstall.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,8 @@ trtllm() {
149149
build
150150
fi
151151

152-
pip install --no-cache-dir $WHEELS_DIR/tensorrt_llm*.whl --extra-index-url https://pypi.nvidia.com || true
152+
pip install --no-cache-dir $WHEELS_DIR/tensorrt_llm*.whl --extra-index-url https://pypi.nvidia.com &&
153+
sed -i '57d' /usr/local/lib/python3.12/dist-packages/torch_tensorrt/dynamo/conversion/custom_ops_converters.py || true
153154
fi
154155
}
155156

tests/export/test_tensorrt_lazy_compiler.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
import tempfile
1818
import unittest
1919
from unittest.mock import MagicMock, patch
20+
2021
import pytest
2122
import torch.nn as nn
2223

0 commit comments

Comments
 (0)