Fix lint

BlueBrain · May 30, 2024 · 6996eeb · 6996eeb
1 parent 3bbe21c
commit 6996eeb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/neurom/core/morphology.py b/neurom/core/morphology.py
@@ -456,7 +456,7 @@ def subtree_types(self):
         for section in it:
             # A subtree can start from a single branch or as a fork of the same type from a parent
             # with different type.
-            if section.type != section.parent.type and section.type != subtree_types[-1]:
+            if section.type not in {section.parent.type, subtree_types[-1]}:
                 subtree_types.append(section.to_morphio().type)
 
         return subtree_types

diff --git a/tests/test_mixed.py b/tests/test_mixed.py
@@ -1143,8 +1143,7 @@ def test_sholl_frequency_pop(mixed_morph):
 
 
 def test_axon_fork_as_axon_carrying_dendrite():
-    """Test that a axon subtree starting as a fork is considered an AcD.
-    """
+    """Test that a axon subtree starting as a fork is considered an AcD."""
     m = neurom.load_morphology(
         """
         1  1   0  0  0   0.5 -1