Updated typing imports im models

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__)
 

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__)
 

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__)