Skip to content

Commit c3a65ef

Browse files
committed
fix: Address some removed imports
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 81d6bcc commit c3a65ef

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

py/torch_tensorrt/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,11 @@
44
import sys
55
from typing import Dict, List
66

7-
from torch_tensorrt._version import (
7+
from torch_tensorrt._version import ( # noqa: F401
88
__cuda_version__,
99
__cudnn_version__,
1010
__tensorrt_version__,
11+
__version__,
1112
)
1213

1314
from packaging import version
@@ -87,6 +88,7 @@ def _find_lib(name: str, paths: List[str]) -> str:
8788
from torch_tensorrt._enums import * # noqa: F403
8889
from torch_tensorrt._Input import Input # noqa: F401
8990
from torch_tensorrt._utils import * # noqa: F403
91+
from torch_tensorrt._utils import sanitized_torch_version
9092

9193
if version.parse(sanitized_torch_version()) >= version.parse("2.1.dev"):
9294
from torch_tensorrt import dynamo # noqa: F401

0 commit comments

Comments
 (0)