Changes in src/sans/fit/Fitting.py [e3efa6b3:5777106] in sasview

File:

• src/sans/fit/Fitting.py (modified) (5 diffs)

src/sans/fit/Fitting.py

@@ -8 +8 @@
 from sans.fit.ScipyFitting import ScipyFit
 from sans.fit.ParkFitting import ParkFit
-from sans.fit.BumpsFitting import BumpsFit
 
-ENGINES={
-    'scipy': ScipyFit,
-    'park': ParkFit,
-    'bumps': BumpsFit,
-}
 
 class Fit(object):
@@ -32 +26 @@
         
     """  
-    def __init__(self, engine='scipy', *args, **kw):
+    def __init__(self, engine='scipy'):
         """
         """
@@ -38 +32 @@
         self._engine = None
         self.fitter_id = None
-        self.set_engine(engine, *args, **kw)
+        self.set_engine(engine)
           
     def __setattr__(self, name, value):
@@ -55 +49 @@
             self.__dict__[name] = value
                 
-    def set_engine(self, word, *args, **kw):
+    def set_engine(self, word):
         """
         Select the type of Fit 
@@ -65 +59 @@
              
         """
-        try:
-            self._engine = ENGINES[word](*args, **kw)
-        except KeyError, exc:
-            raise KeyError("fit engine should be one of scipy, park or bumps")
+        if word == "scipy":
+            self._engine = ScipyFit()
+        elif word == "park":
+            self._engine = ParkFit()
+        else:
+            raise ValueError, "enter the keyword scipy or park"
 
     def fit(self, msg_q=None, q=None, handler=None,