From e46a1211af16336d78032bedbfc4cb32b5c559f8 Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Wed, 12 Jul 2023 18:33:04 +0200
Subject: [PATCH] Updated typing imports im models

---
 src/ptbench/models/alexnet.py  | 3 ++-
 src/ptbench/models/densenet.py | 3 ++-
 src/ptbench/models/pasa.py     | 3 ++-
 3 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/ptbench/models/alexnet.py b/src/ptbench/models/alexnet.py
index c3aadc6d..c567b9b3 100644
--- a/src/ptbench/models/alexnet.py
+++ b/src/ptbench/models/alexnet.py
@@ -13,7 +13,8 @@ import torch.utils.data
 import torchvision.models as models
 import torchvision.transforms
 
-from ..data.typing import Checkpoint, DataLoader, TransformSequence
+from ..data.typing import DataLoader, TransformSequence
+from .typing import Checkpoint
 
 logger = logging.getLogger(__name__)
 
diff --git a/src/ptbench/models/densenet.py b/src/ptbench/models/densenet.py
index 021f6ce2..ea2cab00 100644
--- a/src/ptbench/models/densenet.py
+++ b/src/ptbench/models/densenet.py
@@ -13,7 +13,8 @@ import torch.utils.data
 import torchvision.models as models
 import torchvision.transforms
 
-from ..data.typing import Checkpoint, DataLoader, TransformSequence
+from ..data.typing import DataLoader, TransformSequence
+from .typing import Checkpoint
 
 logger = logging.getLogger(__name__)
 
diff --git a/src/ptbench/models/pasa.py b/src/ptbench/models/pasa.py
index 7f0cf57d..aaa5a2b0 100644
--- a/src/ptbench/models/pasa.py
+++ b/src/ptbench/models/pasa.py
@@ -13,7 +13,8 @@ import torch.optim.optimizer
 import torch.utils.data
 import torchvision.transforms
 
-from ..data.typing import Checkpoint, DataLoader, TransformSequence
+from ..data.typing import DataLoader, TransformSequence
+from .typing import Checkpoint
 
 logger = logging.getLogger(__name__)
 
-- 
GitLab