rewrite the config format

This commit is contained in:
2024-08-24 22:21:33 +03:00
parent 473adc64b8
commit cebfd1a5bf
3 changed files with 160 additions and 203 deletions

View File

@ -13,7 +13,7 @@ from dataclasses import dataclass
from solver import EOLRBSolution, solve_eolrb, create_eolrb_prune_table
from scorer import (
FingerTrickWithRegrip,
load_config,
load_definitions,
generate_finger_tricks,
build_pretty_string_from_finger_tricks_with_regrips,
)
@ -60,7 +60,7 @@ def load_or_generate_prune_table(file_path: str, prune_size: int) -> Dict:
def main():
config = load_config("./moves.yaml")
definitions = load_definitions("./moves.yaml")
prune_table = load_or_generate_prune_table("prune_table.pkl", PRUNE)
print("prune table size:", humanize.naturalsize(sys.getsizeof(prune_table)))
@ -81,7 +81,7 @@ def main():
solutions = get_first_n_from_generator(alg_generator, SOLUTIONS_TO_EVAL)
solutions_with_finger_tricks = [
EOLRBSolutionWithFingertricks(
solution, generate_finger_tricks(config, solution.alg)
solution, generate_finger_tricks(definitions, solution.alg)
)
for solution in solutions
]