Use isort for import formatting

pyproject.toml

@@ -29,6 +29,7 @@ test = [
 dev = [
     "stratocaster[test]",
     "black",
+    "isort",
 ]
 docs = [
      "sphinx",
@@ -42,6 +43,9 @@ requires = [
 ]
 build-backend = "setuptools.build_meta"
 
+[tool.isort]
+profile = "black"
+
 [tool.versioningit]
 default-version = "1+unknown"
 

src/stratocaster/base/__init__.py

@@ -1,2 +1,2 @@
-from .strategy import Strategy, StrategyResult
 from .models import StrategySettings
+from .strategy import Strategy, StrategyResult

src/stratocaster/base/strategy.py

@@ -1,8 +1,8 @@
 import abc
 from typing import Self, TypeVar
 
-from gufe.tokenization import GufeTokenizable, GufeKey
 from gufe import AlchemicalNetwork, ProtocolResult
+from gufe.tokenization import GufeKey, GufeTokenizable
 
 from .models import StrategySettings
 

src/stratocaster/strategies/connectivity.py

@@ -1,11 +1,10 @@
 from gufe import AlchemicalNetwork, ProtocolResult
 from gufe.tokenization import GufeKey
+from pydantic import Field, root_validator, validator
 
 from stratocaster.base import Strategy, StrategyResult
 from stratocaster.base.models import StrategySettings
 
-from pydantic import validator, Field, root_validator
-
 
 # TODO: docstrings
 class ConnectivityStrategySettings(StrategySettings):

src/stratocaster/tests/test_connectivity_strategy.py

@@ -1,16 +1,16 @@
 import math
-from random import shuffle, randint
+from random import randint, shuffle
 
-from gufe import AlchemicalNetwork
 import pytest
+from gufe import AlchemicalNetwork
 from gufe.tests.test_protocol import DummyProtocol, DummyProtocolResult
 
+from stratocaster.base.models import StrategySettings
+from stratocaster.base.strategy import StrategyResult
 from stratocaster.strategies.connectivity import (
     ConnectivityStrategy,
     ConnectivityStrategySettings,
 )
-from stratocaster.base.strategy import StrategyResult
-from stratocaster.base.models import StrategySettings
 from stratocaster.tests.networks import (
     benzene_variants_star_map as _benzene_variants_star_map,
 )

src/stratocaster/tests/test_strategy_abstraction.py

@@ -1,11 +1,10 @@
 import pytest
-
-from stratocaster.base import StrategySettings, Strategy
-from stratocaster.base.strategy import StrategyResult
-
 from gufe import AlchemicalNetwork, ProtocolResult
 from gufe.tokenization import GufeKey
 
+from stratocaster.base import Strategy, StrategySettings
+from stratocaster.base.strategy import StrategyResult
+
 
 class StrategyASettings(StrategySettings):
     pass

src/stratocaster/tests/test_strategy_base.py

@@ -1,7 +1,8 @@
-from stratocaster.base import Strategy, StrategySettings, StrategyResult
 from gufe import AlchemicalNetwork, ProtocolResult
 from gufe.tokenization import GufeKey
 
+from stratocaster.base import Strategy, StrategyResult, StrategySettings
+
 
 class TestStrategyResult: