It's ruff.

src/functionalizer/core.py

@@ -5,10 +5,11 @@
 from pathlib import Path
 
 import pyarrow.parquet as pq
-import sparkmanager as sm
 from fz_td_recipe import Recipe
 from pyspark.sql import functions as F
 
+import sparkmanager as sm
+
 from . import utils
 from .circuit import Circuit
 from .definitions import CheckpointPhases, SortBy

src/functionalizer/filters/definitions.py

@@ -11,7 +11,6 @@
 from pathlib import Path
 
 import sparkmanager as sm
-
 from functionalizer.circuit import Circuit
 from functionalizer.utils import get_logger
 from functionalizer.utils.checkpointing import checkpoint_resume

src/functionalizer/filters/implementations/reduce_and_cut.py

@@ -1,8 +1,8 @@
 """Filter that matches distributions of synapses."""
 
-import sparkmanager as sm
 from pyspark.sql import functions as F
 
+import sparkmanager as sm
 from functionalizer.circuit import touches_per_pathway
 from functionalizer.definitions import CheckpointPhases
 from functionalizer.filters import DatasetOperation, helpers

src/functionalizer/filters/implementations/spine_length.py

@@ -3,9 +3,9 @@
 from operator import attrgetter
 
 import pandas as pd
-import sparkmanager as sm
 from pyspark.sql import functions as F
 
+import sparkmanager as sm
 from functionalizer.filters import DatasetOperation, FilterInitializationError
 from functionalizer.utils import get_logger
 

src/functionalizer/filters/implementations/synapse_properties.py

@@ -1,9 +1,9 @@
 """Filters to add properties to synapses."""
 
-import sparkmanager as sm
 from pyspark.sql import functions as F
 from pyspark.sql import types as T
 
+import sparkmanager as sm
 from functionalizer.filters import DatasetOperation
 from functionalizer.utils import get_logger
 

src/functionalizer/filters/implementations/synapse_reposition.py

@@ -2,8 +2,8 @@
 
 import numpy as np
 import pandas as pd
-import sparkmanager as sm
 
+import sparkmanager as sm
 from functionalizer.filters import DatasetOperation, FilterInitializationError
 
 

src/functionalizer/filters/implementations/touch.py

@@ -2,9 +2,9 @@
 
 import numpy as np
 import pandas as pd
-import sparkmanager as sm
 from pyspark.sql import functions as F
 
+import sparkmanager as sm
 from functionalizer.filters import DatasetOperation
 from functionalizer.utils import get_logger
 

src/functionalizer/io/circuit.py

@@ -9,11 +9,11 @@
 
 import pandas as pd
 import pyarrow.parquet as pq
-import sparkmanager as sm
 from packaging.version import VERSION_PATTERN, Version
 from pyspark.sql import DataFrame
 from pyspark.sql import functions as F
 
+import sparkmanager as sm
 from functionalizer import schema
 from functionalizer.schema import OUTPUT_MAPPING
 from functionalizer.utils import get_logger

src/functionalizer/utils/checkpointing.py

@@ -4,9 +4,10 @@
 from functools import wraps
 from inspect import signature
 
-import sparkmanager as sm
 from pyspark.sql.column import _to_seq
 
+import sparkmanager as sm
+
 from . import get_logger
 from .filesystem import exists, isdir, size
 

src/functionalizer/utils/spark.py

@@ -2,9 +2,10 @@
 
 from contextlib import contextmanager
 
-import sparkmanager as sm
 from pyspark.sql import functions as F
 
+import sparkmanager as sm
+
 from . import get_logger
 
 logger = get_logger(__name__)

tests/conftest.py

@@ -8,6 +8,7 @@
 from pathlib import Path
 
 import pytest
+
 from functionalizer import filters
 from functionalizer.core import Functionalizer
 from functionalizer.definitions import RunningMode as RM

tests/test_data_input_nodes.py

@@ -4,6 +4,7 @@
 
 import pytest
 from conftest import DATADIR, create_functionalizer
+
 from functionalizer.io import NodeData
 
 

tests/test_data_input_parquet.py

@@ -3,6 +3,7 @@
 import numpy
 import pandas as pd
 import pytest
+
 import sparkmanager as sm
 from functionalizer.io.circuit import BRANCH_COLUMNS, EdgeData
 from functionalizer.utils.conf import Configuration

tests/test_data_input_sonata.py

@@ -5,8 +5,9 @@
 import h5py
 import numpy
 import pytest
-import sparkmanager as sm
 from conftest import DATADIR, DEFAULT_ARGS, create_functionalizer
+
+import sparkmanager as sm
 from functionalizer.io.circuit import BRANCH_COLUMNS, EdgeData
 from functionalizer.utils.conf import Configuration
 

tests/test_filters.py

@@ -3,8 +3,9 @@
 import pandas as pd
 import pyspark.sql.functions as F
 import pytest
-import sparkmanager as sm
 from conftest import DATADIR, DEFAULT_ARGS, create_functionalizer
+
+import sparkmanager as sm
 from functionalizer.utils.spark import cache_broadcast_single_part
 
 NUM_AFTER_DISTANCE = 226301

tests/test_gap_junctions.py

@@ -4,9 +4,10 @@
 
 import numpy as np
 import pytest
-from functionalizer.filters import DatasetOperation
 from pyspark.sql import functions as F
 
+from functionalizer.filters import DatasetOperation
+
 # (src, dst), num_connections
 DENDRO_DATA = [
     ((987, 990), 10),  # 6 with exact or abs() == 1 match

tests/test_morpho_shift.py

@@ -4,11 +4,12 @@
 from unittest.mock import MagicMock
 
 import pytest
+from fz_td_recipe import Recipe
+
 import sparkmanager as sm
 from functionalizer.circuit import Circuit
 from functionalizer.schema import LEGACY_MAPPING
 from functionalizer.utils.conf import Configuration
-from fz_td_recipe import Recipe
 
 
 class MockLoader:

tests/test_morphologies.py

@@ -4,6 +4,7 @@
 
 import numpy
 import pytest
+
 from functionalizer.io.morphologies import MorphologyDB
 
 BRANCHES = [

tests/test_property_generation.py

@@ -3,11 +3,12 @@
 from pathlib import Path
 
 import pytest
-import sparkmanager as sm
 from conftest import CIRCUIT_CONFIG, DATADIR
-from functionalizer.filters import DatasetOperation
 from fz_td_recipe import Recipe
 
+import sparkmanager as sm
+from functionalizer.filters import DatasetOperation
+
 
 @pytest.mark.slow
 def test_property_assignment(fz):

tests/test_spine_morphology.py

@@ -4,6 +4,7 @@
 import numpy.testing as npt
 import pandas as pd
 import pytest
+
 from functionalizer.filters.implementations.spine_morphologies import (
     _create_spine_morphology_udf,
     _read_spine_morphology_attributes,