Changeset 0aeba4e in sasview for src/sas/sascalc/fit/BumpsFitting.py

Timestamp:

Apr 16, 2018 4:33:06 PM (6 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

ticket-1094-headless

Children:

f6f3fb4

Parents:

e870702

Message:

code cleanup for bumps fitting wrapper

File:

• src/sas/sascalc/fit/BumpsFitting.py (modified) (5 diffs)

src/sas/sascalc/fit/BumpsFitting.py

@@ -2 +2 @@
 BumpsFitting module runs the bumps optimizer.
 """
+from __future__ import print_function
+
 import os
 from datetime import timedelta, datetime
@@ -9 +11 @@
 
 from bumps import fitters
+
 try:
     from bumps.options import FIT_CONFIG
+    # Preserve bumps default fitter in case someone wants it later
+    BUMPS_DEFAULT_FITTER = FIT_CONFIG.selected_id
     # Default bumps to use the Levenberg-Marquardt optimizer
     FIT_CONFIG.selected_id = fitters.LevenbergMarquardtFit.id
@@ -17 +22 @@
 except ImportError:
     # CRUFT: Bumps changed its handling of fit options around 0.7.5.6
+    # Preserve bumps default fitter in case someone wants it later
+    BUMPS_DEFAULT_FITTER = fitters.FIT_DEFAULT
     # Default bumps to use the Levenberg-Marquardt optimizer
     fitters.FIT_DEFAULT = 'lm'
@@ -126 +133 @@
         if initial_values is not None:
             self._reset_pars(fitted, initial_values)
+        #print("constraints", constraints)
         self.constraints = dict(constraints)
         self.set_fitted(fitted)
@@ -222 +230 @@
     def _setup(self):
         exprs = {}
-        for M in self.models:
-            exprs.update((".".join((M.name, k)), v) for k, v in M.constraints.items())
+        for model in self.models:
+            exprs.update((".".join((model.name, k)), v)
+                         for k, v in model.constraints.items())
         if exprs:
-            symtab = dict((".".join((M.name, k)), p)
-                          for M in self.models
-                          for k, p in M.parameters().items())
+            symtab = dict((".".join((model.name, k)), p)
+                          for model in self.models
+                          for k, p in model.parameters().items())
             self.update = compile_constraints(symtab, exprs)
         else: