Complete list of possible input algorithms

Change-Id: I4bb839344121522243d6f14a77f58438047adc1f
BlueBrain · May 28, 2019 · fa9f488 · fa9f488
1 parent 92202b9
commit fa9f488
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/tns/extract_input/input_parameters.py b/tns/extract_input/input_parameters.py
@@ -1,5 +1,8 @@
 """Input parameters functions"""
 
+tmd_algos = ('tmd', 'tmd_path', 'tmd_gradient', 'tmd_gradient_path',
+             'tmd_apical', 'tmd_apical_path')
+
 
 def parameters(origin=(0., 0., 0.),
                method='trunk',
@@ -21,10 +24,10 @@ def merged_params(data):
         ret = dict()
         if method == 'trunk':
             branching = 'random'
-        elif method in {'tmd', 'tmd_path'}:
+        elif method in tmd_algos:
             branching = 'bio_oriented'
         else:
-            raise KeyError('Method not recognized! Please select trunk, tmd or tmd_path.')
+            raise KeyError('Method not recognized! Please select from: {}.'.format(tmd_algos))
 
         ret.update({"randomness": 0.15,
                     "targeting": 0.12,

diff --git a/tns/version.py b/tns/version.py
@@ -1,2 +1,2 @@
 """ tns version """
-VERSION = "1.0.10.dev0"
+VERSION = "1.0.11.dev0"