Update import order

tests/core_tests/test_create.py

@@ -1,9 +1,10 @@
 from __future__ import annotations
 
+import sys
+
 import pytest
-from utmosv2._core.create import create_model
 
-import sys
+from utmosv2._core.create import create_model
 
 
 @pytest.mark.parametrize(

train.py

@@ -4,11 +4,11 @@
 
 import numpy as np
 import torch
-from utmosv2._settings._config import Config
 import wandb
 from dotenv import load_dotenv
 
 from utmosv2._settings import configure_args, configure_defaults
+from utmosv2._settings._config import Config
 from utmosv2.runner import run_train
 from utmosv2.utils import (
     get_dataloader,

utmosv2/_core/model/_models.py

@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+from typing import TYPE_CHECKING, Any
 
 from utmosv2._core.model._common import UTMOSv2ModelMixin
 from utmosv2._settings._config import Config
@@ -10,7 +11,6 @@
     SSLMultiSpecExtModelV1,
     SSLMultiSpecExtModelV2,
 )
-from typing import TYPE_CHECKING, Any
 
 if TYPE_CHECKING:
     import torch

utmosv2/_settings/_config.py

@@ -1,8 +1,8 @@
 from __future__ import annotations
 
 import argparse
-from pathlib import Path
 import sys
+from pathlib import Path
 from types import ModuleType, SimpleNamespace
 
 if sys.version_info >= (3, 10):
@@ -13,6 +13,7 @@
     Config: TypeAlias = SimpleNamespace | ModuleType
 else:
     from typing import Union
+
     from typing_extensions import TypeAlias
 
     Config: TypeAlias = Union[SimpleNamespace, ModuleType]

utmosv2/model/multi_spec.py

@@ -1,8 +1,9 @@
+from typing import cast
+
 import timm
 import torch
 import torch.nn as nn
 import torch.nn.functional as F
-from typing import cast
 
 from utmosv2._settings._config import Config
 from utmosv2.dataset._utils import get_dataset_num

utmosv2/model/ssl_multispec.py

@@ -1,6 +1,7 @@
+from typing import cast
+
 import torch
 import torch.nn as nn
-from typing import cast
 
 from utmosv2._settings._config import Config
 from utmosv2.dataset._utils import get_dataset_num