langchain/libs/partners/nvidia-trt/tests/unit_tests/test_imports.py

8 lines
164 B
Python
Raw Normal View History

from langchain_nvidia_trt import __all__
EXPECTED_ALL = ["TritonTensorRTLLM"]
def test_all_imports() -> None:
assert sorted(EXPECTED_ALL) == sorted(__all__)