Skip to content

Commit 03a1765

Browse files
committed
Fix types for load_dataframe
1 parent a615ec4 commit 03a1765

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

adaptive/learner/sequence_learner.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from sortedcontainers import SortedDict, SortedSet
88

99
from adaptive.learner.base_learner import BaseLearner
10+
from adaptive.types import Int
1011
from adaptive.utils import assign_defaults, partial_function_from_dataframe
1112

1213
try:
@@ -23,7 +24,7 @@
2324
from typing_extensions import TypeAlias
2425

2526

26-
PointType: TypeAlias = Tuple[int, Any]
27+
PointType: TypeAlias = Tuple[Int, Any]
2728

2829

2930
class _IgnoreFirstArgument:
@@ -224,7 +225,9 @@ def load_dataframe(
224225
y_name : str, optional
225226
The ``y_name`` used in ``to_dataframe``, by default "y"
226227
"""
227-
self.tell_many(df[[index_name, x_name]].values, df[y_name].values)
228+
indices = df[index_name].values
229+
xs = df[x_name].values
230+
self.tell_many(zip(indices, xs), df[y_name].values)
228231
if with_default_function_args:
229232
self.function = partial_function_from_dataframe(
230233
self._original_function, df, function_prefix

0 commit comments

Comments
 (0)