Skip to content

Commit 604cc19

Browse files
narendasanlaikhtewari
authored andcommitted
chore: small mypy issue (#2803)
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 61b5280 commit 604cc19

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

py/torch_tensorrt/dynamo/_settings.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
from dataclasses import dataclass, field
2-
from typing import Collection, Optional, Union
2+
from typing import Collection, Optional, Set, Union
33

44
from torch.fx.node import Target
55
from torch_tensorrt._Device import Device
@@ -71,7 +71,7 @@ class CompilationSettings:
7171
hardware_compatible (bool): Build the TensorRT engines compatible with GPU architectures other than that of the GPU on which the engine was built (currently works for NVIDIA Ampere and newer)
7272
"""
7373

74-
enabled_precisions: dtype = field(default_factory=lambda: ENABLED_PRECISIONS)
74+
enabled_precisions: Set[dtype] = field(default_factory=lambda: ENABLED_PRECISIONS)
7575
debug: bool = DEBUG
7676
workspace_size: int = WORKSPACE_SIZE
7777
min_block_size: int = MIN_BLOCK_SIZE

0 commit comments

Comments
 (0)