speed up len() and .shape for neuronlists

navis-org · Jun 27, 2018 · 3a044cf · 3a044cf
1 parent 65be9f2
commit 3a044cf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pymaid/core.py b/pymaid/core.py
@@ -1428,7 +1428,8 @@ def __next__(self):
         return prange_iter(self.neurons, 0)
 
     def __len__(self):
-        return len(self.neurons)
+        """Use skeleton ID here, otherwise this is terribly slow."""
+        return len(self.skeleton_id)
 
     def __getattr__(self, key):
         if key == 'shape':