Skip to content

Commit 75472f0

Browse files
committed
Use _set_data in Learner2D.load_dataframe
1 parent d859c45 commit 75472f0

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

adaptive/learner/learner2D.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -474,7 +474,8 @@ def load_dataframe(
474474
z_name : str, optional
475475
The ``z_name`` used in ``to_dataframe``, by default "z"
476476
"""
477-
self.tell_many(df[[x_name, y_name]].values, df[z_name].values)
477+
data = df.set_index([x_name, y_name])[z_name].to_dict()
478+
self._set_data(data)
478479
if with_default_function_args:
479480
self.function = partial_function_from_dataframe(
480481
self.function, df, function_prefix

0 commit comments

Comments
 (0)