neurom.core.types.tree_type_checker¶

neurom.core.types.tree_type_checker(*ref)[source]¶

Tree type checker functor

Returns:	Functor that takes a tree, and returns true if that tree matches any of NeuriteTypes in ref

Ex:

>>> from neurom.core.types import NeuriteType, tree_type_checker
>>> tree_filter = tree_type_checker(NeuriteType.axon, NeuriteType.basal_dendrite)
>>> nrn.i_neurites(tree.isegment, tree_filter=tree_filter)