estimators sorted

This commit is contained in:
Lorenzo Volpi 2023-12-02 02:06:53 +01:00
parent ba818157db
commit 86025a804c
1 changed files with 2 additions and 1 deletions

View File

@ -342,10 +342,11 @@ def update_estimators(href, dataset, metric, curr_estimators, root):
old_estimators = json.loads(old_estimators) old_estimators = json.loads(old_estimators)
except JSONDecodeError: except JSONDecodeError:
old_estimators = [] old_estimators = []
valid_estimators = dr.data(metric=metric).columns.unique(0).to_numpy() valid_estimators: np.ndarray = dr.data(metric=metric).columns.unique(0).to_numpy()
new_estimators = valid_estimators[ new_estimators = valid_estimators[
np.isin(valid_estimators, old_estimators) np.isin(valid_estimators, old_estimators)
].tolist() ].tolist()
valid_estimators = CE.name.sort(valid_estimators.tolist())
return valid_estimators, new_estimators return valid_estimators, new_estimators