neurom.core.types.tree_type_checker

neurom.core.types.tree_type_checker(*ref)

Tree type checker functor

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

Ex:

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