Skip to content

Commit 8723e6c

Browse files
Joao Gomesfacebook-github-bot
authored andcommitted
[fbsync] Fix MaxViT typing (#6929)
Summary: * Fix MaxViT typing * linter Reviewed By: YosuaMichael Differential Revision: D41376283 fbshipit-source-id: 1f37bc6b3f6bd2813dcc64c29a29e1f68d4444df
1 parent a855b97 commit 8723e6c

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)