Fix pre-commits complaints

{{ cookiecutter.repository_name }}/src/{{ cookiecutter.package_name }}/data/datamodule.py

@@ -4,8 +4,8 @@
 from typing import List, Mapping, Optional
 
 import hydra
-import omegaconf
 import lightning.pytorch as pl
+import omegaconf
 from omegaconf import DictConfig
 from torch.utils.data import DataLoader, Dataset
 from torch.utils.data.dataloader import default_collate

{{ cookiecutter.repository_name }}/src/{{ cookiecutter.package_name }}/pl_modules/pl_module.py

@@ -2,8 +2,8 @@
 from typing import Any, Dict, Mapping, Optional, Sequence, Tuple, Union
 
 import hydra
-import omegaconf
 import lightning.pytorch as pl
+import omegaconf
 import torch
 import torch.nn.functional as F
 import torchmetrics

{{ cookiecutter.repository_name }}/src/{{ cookiecutter.package_name }}/run.py

@@ -2,11 +2,11 @@
 from typing import List, Optional
 
 import hydra
-import omegaconf
 import lightning.pytorch as pl
+import omegaconf
 import torch
-from omegaconf import DictConfig, ListConfig
 from lightning.pytorch import Callback
+from omegaconf import DictConfig, ListConfig
 
 from nn_core.callbacks import NNTemplateCore
 from nn_core.common import PROJECT_ROOT

{{ cookiecutter.repository_name }}/tests/conftest.py

@@ -7,9 +7,9 @@
 import pytest
 from hydra import compose, initialize
 from hydra.core.hydra_config import HydraConfig
+from lightning.pytorch import seed_everything
 from omegaconf import DictConfig, OmegaConf, open_dict
 from pytest import FixtureRequest, TempPathFactory
-from lightning.pytorch import seed_everything
 
 from nn_core.serialization import NNCheckpointIO
 

{{ cookiecutter.repository_name }}/tests/test_checkpoint.py

@@ -2,9 +2,9 @@
 from pathlib import Path
 from typing import Any, Dict
 
-from omegaconf import DictConfig, OmegaConf
 from lightning.pytorch import LightningModule
 from lightning.pytorch.core.saving import _load_state
+from omegaconf import DictConfig, OmegaConf
 
 from nn_core.serialization import NNCheckpointIO
 from tests.conftest import load_checkpoint