Skip to content

Commit c18ecc4

Browse files
committed
Fix _get_data and _set_data types
1 parent 0c6f1e5 commit c18ecc4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

adaptive/learner/average_learner1D.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -577,10 +577,10 @@ def tell_many_at_point(self, x: Real, seed_y_mapping: dict[int, Real]) -> None:
577577
self._update_interpolated_loss_in_interval(*interval)
578578
self._oldscale = deepcopy(self._scale)
579579

580-
def _get_data(self) -> dict[Real, Real]:
580+
def _get_data(self) -> dict[Real, dict[Int, Real]]:
581581
return self._data_samples
582582

583-
def _set_data(self, data: dict[Real, Real]) -> None:
583+
def _set_data(self, data: dict[Real, dict[Int, Real]]) -> None:
584584
if data:
585585
for x, samples in data.items():
586586
self.tell_many_at_point(x, samples)

0 commit comments

Comments
 (0)