Fix tests #43

opengridcc · May 14, 2018 · bbfbee4 · bbfbee4
1 parent 4f146f5
commit bbfbee4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/opengrid/library/regression.py b/opengrid/library/regression.py
@@ -502,8 +502,8 @@ def __getstate__(self):
         for fit in self._list_of_fits:
                 d['formulas'].append(self._modeldesc_to_dict(fit.model.formula))
                 #delattr(fit.model, 'formula')
-        d.pop('list_of_fits')
-        d.pop('fit')
+        d.pop('_list_of_fits')
+        d.pop('_fit')
 
         print("Pickling...  Removing the 'formula' from each fit.model.\n\
              You have to unpickle your object or run __setstate__(self.__dict__) to restore them.".format(d))