from .base import check_random_state