STY: Added docstring to BenchLeastSquares + PEP8 fixes

scipy · May 25, 2015 · 3b3b678 · 3b3b678
1 parent 6107403
commit 3b3b678
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/benchmarks/benchmarks/lsq_problems.py b/benchmarks/benchmarks/lsq_problems.py
@@ -107,7 +107,7 @@ def check_answer(self, x, ftol):
             then False is returned.
         """
         if (self.lb is not None and np.any(x < self.lb) or
-            self.ub is not None and np.any(x > self.ub)):
+                self.ub is not None and np.any(x > self.ub)):
             return False
 
         f = np.sum(self.fun(x) ** 2)

diff --git a/benchmarks/benchmarks/optimize.py b/benchmarks/benchmarks/optimize.py
@@ -8,13 +8,10 @@
 try:
     import scipy.optimize
     from scipy.optimize.optimize import rosen, rosen_der, rosen_hess
+    from scipy.optimize import leastsq
 except ImportError:
     pass
 
-from scipy.optimize import leastsq
-
-
-
 from . import test_functions as funcs
 from .common import Benchmark
 from .lsq_problems import extract_lsq_problems
@@ -242,6 +239,7 @@ def run_LJ(self, methods=None):
 
 
 class BenchLeastSquares(Benchmark):
+    """Class for benchmarking nonlinear least squares solvers."""
     problems = extract_lsq_problems()
     params = [
         list(problems.keys()),