rewrite the config format
This commit is contained in:
@ -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
|
||||
]
|
||||
|
||||
Reference in New Issue
Block a user