Unverified Commit 3f4dcae6 authored by Vasilis Vryniotis's avatar Vasilis Vryniotis Committed by GitHub
Browse files

Fix MaxViT typing (#6929)

* Fix MaxViT typing

* linter
parent ad128b75
import math import math
from collections import OrderedDict
from functools import partial from functools import partial
from typing import Any, Callable, List, Optional, OrderedDict, Sequence, Tuple from typing import Any, Callable, List, Optional, Sequence, Tuple
import numpy as np import numpy as np
import torch import torch
...@@ -426,7 +427,7 @@ class MaxVitLayer(nn.Module): ...@@ -426,7 +427,7 @@ class MaxVitLayer(nn.Module):
) -> None: ) -> None:
super().__init__() super().__init__()
layers: OrderedDict[str, Any] = OrderedDict() # type: ignore layers: OrderedDict = OrderedDict()
# convolutional layer # convolutional layer
layers["MBconv"] = MBConv( layers["MBconv"] = MBConv(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment