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
from collections import OrderedDict
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 torch
......@@ -426,7 +427,7 @@ class MaxVitLayer(nn.Module):
) -> None:
super().__init__()
layers: OrderedDict[str, Any] = OrderedDict() # type: ignore
layers: OrderedDict = OrderedDict()
# convolutional layer
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