split_axon_dendrite: fix typing

navis-org · Nov 16, 2020 · e28f5c2 · e28f5c2
1 parent 5c556ea
commit e28f5c2
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/navis/morpho/manipulation.py b/navis/morpho/manipulation.py
@@ -328,11 +328,9 @@ def prune_twigs(x: NeuronObject,
 
 
 def split_axon_dendrite(x: NeuronObject,
-                        metric: Union[Literal['centrifugal'],
-                                      Literal['centripetal'],
-                                      Literal['sum'],
-                                      Literal['bending'],
-                                      Literal['segregation']] = 'segregation',
+                        metric: Union[Literal['flow_centrality'],
+                                      Literal['bending_flow'],
+                                      Literal['segregation_index']] = 'flow_centrality',
                         flow_thresh: float = .9,
                         split: Union[Literal['prepost'],
                                      Literal['distance']] = 'prepost',