BUG: pass on `nsim` to other param_evaluator

gyral_structure/basis/core.py

@@ -523,7 +523,7 @@ class SumBase(UserList):
         """
         self._fixed = {}
 
-    def param_evaluator(self, parameters, method=None, extent=0):
+    def param_evaluator(self, parameters, method=None, extent=0, nsim=2**12):
         """Returns a function that evaluates the basis function for arbitrary positions given a fixed parameter array
 
         Used to __call__ streamlines
@@ -531,16 +531,17 @@ class SumBase(UserList):
         :param parameters: (nparams, ) array defining the parameters for which the field will be evaluated
         :param method: Algorithm used to __call__ the basis functions
         :param extent: maximum extent of request to accept (keep at zero for tractography)
+        :param nsim: maximum number of positions to evaluate simultaneously
         :return: function that maps positions to vector field
         """
         funcs = []
         idx_param = 0
         for idx_elem, elem in enumerate(self):
             if idx_elem in self._fixed:
-                funcs.append(elem.param_evaluator(self._fixed[idx_elem], extent=extent, method=method))
+                funcs.append(elem.param_evaluator(self._fixed[idx_elem], extent=extent, method=method, nsim=nsim))
             else:
                 funcs.append(elem.param_evaluator(parameters[idx_param:idx_param + elem.nparams],
-                                                  extent=extent, method=method))
+                                                  extent=extent, method=method, nsim=nsim))
                 idx_param += elem.nparams
 
         def evaluate(positions):