diff options
| -rwxr-xr-x | src/caelestia/utils/material/score.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/caelestia/utils/material/score.py b/src/caelestia/utils/material/score.py index da8b062..7765050 100755 --- a/src/caelestia/utils/material/score.py +++ b/src/caelestia/utils/material/score.py @@ -20,9 +20,8 @@ class Score: pass @staticmethod - def score(colors_to_population: dict) -> tuple[list[Hct], list[Hct]]: + def score(colors_to_population: dict, filter_enabled: bool = True) -> tuple[list[Hct], list[Hct]]: desired = 14 - filter_enabled = True dislike_filter = True colors_hct = [] @@ -110,6 +109,10 @@ class Score: for i, chosen_hct in enumerate(chosen_colors): chosen_colors[i] = DislikeAnalyzer.fix_if_disliked(chosen_hct) + # Ensure enough colours + if len(chosen_colors) < desired: + return Score.score(colors_to_population, False) + return chosen_primaries, chosen_colors |