CubePool split by color category

This commit is contained in:
henriquenakashima 2021-02-22 21:26:11 -05:00
parent cba1008b17
commit b57ddacefd
3 changed files with 67 additions and 9 deletions

View file

@ -21,8 +21,9 @@ def main():
pools = cube_pool.load_pools(csv_path, {'core', 'occasional'})
main_pool = (random.sample(pools['core'], CARDS_FROM_CORE) +
random.sample(pools['occasional'], CARDS_FROM_OCCASIONAL))
main_pool = (pools['core'].wide_sample(CARDS_FROM_CORE) +
pools['occasional'].wide_sample(CARDS_FROM_OCCASIONAL))
with open(OUTPUT_FILENAME, 'w') as f:
f.write('\n'.join(main_pool))