Use self._original_function when loading from dataframe

basnijholt · basnijholt · commit 777c6d040383 · 2022-10-10T19:44:49.000-07:00
diff --git a/adaptive/learner/sequence_learner.py b/adaptive/learner/sequence_learner.py
@@ -216,7 +216,7 @@ def load_dataframe(
         self.tell_many(df[[index_name, x_name]].values, df[y_name].values)
         if with_default_function_args:
             self.function = partial_function_from_dataframe(
-                self.function, df, function_prefix
+                self._original_function, df, function_prefix
             )
 
     def _get_data(self):

Original file line number	Diff line number	Diff line change
`@@ -216,7 +216,7 @@ def load_dataframe(`
`216`	`216`	`self.tell_many(df[[index_name, x_name]].values, df[y_name].values)`
`217`	`217`	`if with_default_function_args:`
`218`	`218`	`self.function = partial_function_from_dataframe(`
`219`		`- self.function, df, function_prefix`
	`219`	`+ self._original_function, df, function_prefix`
`220`	`220`	`)`
`221`	`221`
`222`	`222`	`def _get_data(self):`