@@ -108,12 +108,12 @@ def __init__(
108
108
self ._undersampled_points : Set [Real ] = set ()
109
109
# Contains the error in the estimate of the
110
110
# mean at each point x in the form {x0: error(x0), ...}
111
- self .error : ItemSortedDict [Real , float ] = decreasing_dict ()
111
+ self .error : Dict [Real , float ] = decreasing_dict ()
112
112
# Distance between two neighboring points in the
113
113
# form {xi: ((xii-xi)^2 + (yii-yi)^2)^0.5, ...}
114
- self ._distances : ItemSortedDict [Real , float ] = decreasing_dict ()
114
+ self ._distances : Dict [Real , float ] = decreasing_dict ()
115
115
# {xii: error[xii]/min(_distances[xi], _distances[xii], ...}
116
- self .rescaled_error : ItemSortedDict [Real , float ] = decreasing_dict ()
116
+ self .rescaled_error : Dict [Real , float ] = decreasing_dict ()
117
117
118
118
@property
119
119
def nsamples (self ) -> int :
@@ -444,10 +444,10 @@ def tell_many_at_point(self, x: Real, seed_y_mapping: Dict[int, Real]) -> None:
444
444
self ._update_interpolated_loss_in_interval (* interval )
445
445
self ._oldscale = deepcopy (self ._scale )
446
446
447
- def _get_data (self ) -> SortedDict [Real , Real ]:
447
+ def _get_data (self ) -> Dict [Real , Real ]:
448
448
return self ._data_samples
449
449
450
- def _set_data (self , data : SortedDict [Real , Real ]) -> None :
450
+ def _set_data (self , data : Dict [Real , Real ]) -> None :
451
451
if data :
452
452
for x , samples in data .items ():
453
453
self .tell_many_at_point (x , samples )
@@ -481,7 +481,7 @@ def plot(self):
481
481
return p .redim (x = dict (range = plot_bounds ))
482
482
483
483
484
- def decreasing_dict () -> ItemSortedDict :
484
+ def decreasing_dict () -> Dict :
485
485
"""This initialization orders the dictionary from large to small values"""
486
486
487
487
def sorting_rule (key , value ):
0 commit comments