From f5c4d5cbb7346dff67b94742c20584d5eed98a80 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sat, 11 Jan 2025 15:13:16 +0100 Subject: [PATCH] Add missing imports --- timm/layers/attention2d.py | 2 +- timm/models/fastvit.py | 2 +- timm/models/swin_transformer_v2.py | 2 +- timm/models/vgg.py | 2 +- 4 files changed, 4 insertions(+), 4 deletions(-) diff --git a/timm/layers/attention2d.py b/timm/layers/attention2d.py index f56cd84e..e5601aec 100644 --- a/timm/layers/attention2d.py +++ b/timm/layers/attention2d.py @@ -1,4 +1,4 @@ -from typing import List, Optional, Union +from typing import List, Optional, Type, Union import torch from torch import nn as nn diff --git a/timm/models/fastvit.py b/timm/models/fastvit.py index 3bc17e2d..1e821810 100644 --- a/timm/models/fastvit.py +++ b/timm/models/fastvit.py @@ -7,7 +7,7 @@ # import os from functools import partial -from typing import List, Optional, Tuple, Union +from typing import List, Optional, Tuple, Type, Union import torch import torch.nn as nn diff --git a/timm/models/swin_transformer_v2.py b/timm/models/swin_transformer_v2.py index db5486bd..6a3330f4 100644 --- a/timm/models/swin_transformer_v2.py +++ b/timm/models/swin_transformer_v2.py @@ -13,7 +13,7 @@ Modifications and additions for timm hacked together by / Copyright 2022, Ross W # Written by Ze Liu # -------------------------------------------------------- import math -from typing import Callable, List, Optional, Tuple, Union +from typing import Callable, List, Optional, Tuple, Type, Union import torch import torch.nn as nn diff --git a/timm/models/vgg.py b/timm/models/vgg.py index 7c140a87..c096df23 100644 --- a/timm/models/vgg.py +++ b/timm/models/vgg.py @@ -5,7 +5,7 @@ timm functionality. Copyright 2021 Ross Wightman """ -from typing import Any, Dict, List, Optional, Union, cast +from typing import Any, Dict, List, Optional, Type, Union, cast import torch import torch.nn as nn