randomizer.py

1from typing import List2from mrlypy.core.state import choice3from .enums import Design45def random_design(designs: List[Design] = None) -> Design:6    if designs is None:7        designs = list(Design)8    return choice(designs)910def random_rotation(design: Design) -> int:11    return choice([0, 1]) if design == Design.TREE else 0