Skip to content

Commit 3f4dcae

Browse files
authored
Fix MaxViT typing (#6929)
* Fix MaxViT typing * linter
1 parent ad128b7 commit 3f4dcae

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

torchvision/models/maxvit.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import math
2+
from collections import OrderedDict
23
from functools import partial
3-
from typing import Any, Callable, List, Optional, OrderedDict, Sequence, Tuple
4+
from typing import Any, Callable, List, Optional, Sequence, Tuple
45

56
import numpy as np
67
import torch
@@ -426,7 +427,7 @@ def __init__(
426427
) -> None:
427428
super().__init__()
428429

429-
layers: OrderedDict[str, Any] = OrderedDict() # type: ignore
430+
layers: OrderedDict = OrderedDict()
430431

431432
# convolutional layer
432433
layers["MBconv"] = MBConv(

0 commit comments

Comments
 (0)