Lint.

src/functionalizer/filters/definitions.py

@@ -39,7 +39,7 @@ def load(*dirnames: str) -> None:
 
 
 class FilterInitializationError(RuntimeError):
-    pass
+    """Error to be raised when filters should be skipped."""
 
 
 # ---------------------------------------------------

tests/test_data_input_sonata.py

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

tests/test_filters.py

@@ -4,7 +4,7 @@
 import pyspark.sql.functions as F
 import pytest
 import sparkmanager as sm
-from conftest import DEFAULT_ARGS, DATADIR, create_functionalizer
+from conftest import DATADIR, DEFAULT_ARGS, create_functionalizer
 from functionalizer.utils.spark import cache_broadcast_single_part
 
 NUM_AFTER_DISTANCE = 226301
@@ -57,17 +57,13 @@ def test_spine_morphos(self, fz):
 class TestBogusFilterInitialization:
     """Test initialization of optional filters"""
 
-    @pytest.mark.parametrize("fz", [{"circuit_config": DATADIR / "circuit_config_invalid.json"}], indirect=True)
+    @pytest.mark.parametrize(
+        "fz", [{"circuit_config": DATADIR / "circuit_config_invalid.json"}], indirect=True
+    )
     def test_spine_morphos(self, fz):
         with pytest.raises(AssertionError):
             fz.process_filters(filters=["SpineMorphologies"])
 
-class TestFilterInitialization:
-    """Test initialization of optional filters"""
-
-    def test_spine_morphos(self, fz):
-        fz.process_filters(filters=["SpineMorphologies"])
-
 
 @pytest.mark.slow
 class TestFilters(object):