Change to known first party

pyproject.toml

@@ -37,4 +37,4 @@ count = true
 skips = ["B101", "B311"]
 
 [tool.isort]
-known_third_party = ["tests"]
+known_first_party = ["tests", "src"]

runner/src/datamodules/distribution_datamodule.py

@@ -7,10 +7,11 @@
 from pytorch_lightning import LightningDataModule
 from pytorch_lightning.trainer.supporters import CombinedLoader
 from sklearn.preprocessing import StandardScaler
-from src import utils
 from torch.utils.data import DataLoader, Sampler, TensorDataset, random_split
 from torchdyn.datasets import ToyDataset
 
+from src import utils
+
 from .components.base import BaseLightningDataModule
 from .components.time_dataset import load_dataset
 from .components.tnet_dataset import SCData

runner/src/eval.py

@@ -38,6 +38,7 @@
 from omegaconf import DictConfig
 from pytorch_lightning import LightningDataModule, LightningModule, Trainer
 from pytorch_lightning.loggers import LightningLoggerBase
+
 from src import utils
 
 log = utils.get_pylogger(__name__)

runner/src/train.py

@@ -39,6 +39,7 @@
 from omegaconf import DictConfig
 from pytorch_lightning import Callback, LightningDataModule, LightningModule, Trainer
 from pytorch_lightning.loggers import LightningLoggerBase
+
 from src import utils
 
 log = utils.get_pylogger(__name__)

runner/src/utils/rich_utils.py

@@ -8,6 +8,7 @@
 from omegaconf import DictConfig, OmegaConf, open_dict
 from pytorch_lightning.utilities import rank_zero_only
 from rich.prompt import Prompt
+
 from src.utils import pylogger
 
 log = pylogger.get_pylogger(__name__)

runner/src/utils/utils.py

@@ -9,6 +9,7 @@
 from pytorch_lightning import Callback
 from pytorch_lightning.loggers import LightningLoggerBase
 from pytorch_lightning.utilities import rank_zero_only
+
 from src.utils import pylogger, rich_utils
 
 log = pylogger.get_pylogger(__name__)

runner/tests/helpers/run_if.py

@@ -10,6 +10,7 @@
 import torch
 from packaging.version import Version
 from pkg_resources import get_distribution
+
 from tests.helpers.package_available import (
     _COMET_AVAILABLE,
     _DEEPSPEED_AVAILABLE,

runner/tests/helpers/run_sh_command.py

@@ -1,6 +1,7 @@
 from typing import List
 
 import pytest
+
 from tests.helpers.package_available import _SH_AVAILABLE
 
 if _SH_AVAILABLE:

runner/tests/test_datamodule.py

@@ -2,6 +2,7 @@
 
 import pytest
 import torch
+
 from src.datamodules.distribution_datamodule import (
     SKLearnDataModule,
     TorchDynDataModule,

runner/tests/test_eval.py

@@ -3,6 +3,7 @@
 import pytest
 from hydra.core.hydra_config import HydraConfig
 from omegaconf import open_dict
+
 from src.eval import evaluate
 from src.train import train
 

runner/tests/test_sweeps.py

@@ -1,4 +1,5 @@
 import pytest
+
 from tests.helpers.run_if import RunIf
 from tests.helpers.run_sh_command import run_sh_command
 

runner/tests/test_train.py

@@ -3,6 +3,7 @@
 import pytest
 from hydra.core.hydra_config import HydraConfig
 from omegaconf import open_dict
+
 from src.train import train
 from tests.helpers.run_if import RunIf