| from dataclasses import dataclass, field |
|
|
| try: |
| import trackio.utils as utils |
| except ImportError: |
| import utils |
|
|
|
|
| @dataclass |
| class RunSelection: |
| choices: list[str] = field(default_factory=list) |
| selected: list[str] = field(default_factory=list) |
| locked: bool = False |
|
|
| def update_choices( |
| self, runs: list[str], preferred: list[str] | None = None |
| ) -> bool: |
| if self.choices == runs: |
| return False |
| new_choices = set(runs) - set(self.choices) |
| self.choices = list(runs) |
| if self.locked: |
| base = set(self.selected) | new_choices |
| elif preferred: |
| base = set(preferred) |
| else: |
| base = set(runs) |
| self.selected = [run for run in self.choices if run in base] |
| return True |
|
|
| def select(self, runs: list[str]) -> list[str]: |
| choice_set = set(self.choices) |
| self.selected = [run for run in runs if run in choice_set] |
| self.locked = True |
| return self.selected |
|
|
| def replace_group( |
| self, group_runs: list[str], new_subset: list[str] | None |
| ) -> tuple[list[str], list[str]]: |
| new_subset = utils.ordered_subset(group_runs, new_subset) |
| selection_set = set(self.selected) |
| selection_set.difference_update(group_runs) |
| selection_set.update(new_subset) |
| self.selected = [run for run in self.choices if run in selection_set] |
| self.locked = True |
| return new_subset, self.selected |
|
|
