adopting to new prospector

This commit is contained in:
Martin Diehl 2019-09-19 11:44:15 -07:00
parent 35a009f513
commit 2e25a03d13
3 changed files with 10 additions and 12 deletions

View File

@ -1,6 +1,4 @@
# -*- coding: UTF-8 no BOM -*- """Main aggregator."""
"""Main aggregator"""
import os import os
with open(os.path.join(os.path.dirname(__file__),'VERSION')) as f: with open(os.path.join(os.path.dirname(__file__),'VERSION')) as f:

View File

@ -1,5 +1,3 @@
# -*- coding: UTF-8 no BOM -*- """Test functionality."""
"""Test functionality"""
from .test import Test # noqa from .test import Test # noqa

View File

@ -279,6 +279,7 @@ def leastsqBound(func, x0, args=(), bounds=None, Dfun=None, full_output=0,
def _int2extFunc(bounds): def _int2extFunc(bounds):
"""Transform internal parameters into external parameters.""" """Transform internal parameters into external parameters."""
local = [_int2extLocal(b) for b in bounds] local = [_int2extLocal(b) for b in bounds]
def _transform_i2e(p_int): def _transform_i2e(p_int):
p_ext = np.empty_like(p_int) p_ext = np.empty_like(p_int)
p_ext[:] = [i(j) for i, j in zip(local, p_int)] p_ext[:] = [i(j) for i, j in zip(local, p_int)]
@ -288,6 +289,7 @@ def leastsqBound(func, x0, args=(), bounds=None, Dfun=None, full_output=0,
def _ext2intFunc(bounds): def _ext2intFunc(bounds):
"""Transform external parameters into internal parameters.""" """Transform external parameters into internal parameters."""
local = [_ext2intLocal(b) for b in bounds] local = [_ext2intLocal(b) for b in bounds]
def _transform_e2i(p_ext): def _transform_e2i(p_ext):
p_int = np.empty_like(p_ext) p_int = np.empty_like(p_ext)
p_int[:] = [i(j) for i, j in zip(local, p_ext)] p_int[:] = [i(j) for i, j in zip(local, p_ext)]
@ -427,7 +429,7 @@ def _weighted_general_function(params, ydata, xdata, function, weights):
return (function(xdata, *params) - ydata)*weights return (function(xdata, *params) - ydata)*weights
def curve_fit_bound(f, xdata, ydata, p0=None, sigma=None, bounds=None, **kw): def curve_fit_bound(f, xdata, ydata, p0=None, sigma=None, bounds=None, **kw):
"""Similar as 'curve_fit' in minpack.py""" """Similar as 'curve_fit' in minpack.py."""
if p0 is None: if p0 is None:
# determine number of parameters by inspecting the function # determine number of parameters by inspecting the function
import inspect import inspect
@ -467,7 +469,7 @@ def curve_fit_bound(f, xdata, ydata, p0=None, sigma=None, bounds=None, **kw):
return (popt, pcov, infodict, errmsg, ier) if return_full else (popt, pcov) return (popt, pcov, infodict, errmsg, ier) if return_full else (popt, pcov)
class Worker(Thread): class Worker(Thread):
"""Thread executing tasks from a given tasks queue""" """Thread executing tasks from a given tasks queue."""
def __init__(self, tasks): def __init__(self, tasks):
Thread.__init__(self) Thread.__init__(self)
@ -489,7 +491,7 @@ class Worker(Thread):
class ThreadPool: class ThreadPool:
"""Pool of threads consuming tasks from a queue""" """Pool of threads consuming tasks from a queue."""
def __init__(self, num_threads): def __init__(self, num_threads):
self.tasks = Queue(num_threads) self.tasks = Queue(num_threads)
@ -497,14 +499,14 @@ class ThreadPool:
Worker(self.tasks) Worker(self.tasks)
def add_task(self, func, *args, **kargs): def add_task(self, func, *args, **kargs):
"""Add a task to the queue""" """Add a task to the queue."""
self.tasks.put((func, args, kargs)) self.tasks.put((func, args, kargs))
def map(self, func, args_list): def map(self, func, args_list):
"""Add a list of tasks to the queue""" """Add a list of tasks to the queue."""
for args in args_list: for args in args_list:
self.add_task(func, args) self.add_task(func, args)
def wait_completion(self): def wait_completion(self):
"""Wait for completion of all the tasks in the queue""" """Wait for completion of all the tasks in the queue."""
self.tasks.join() self.tasks.join()