Source code for ants.config

_deterministic = False
_random_seed = None



def set_ants_deterministic(on=True, seed_value=123):
    """
    Set deterministic behavior globally for the package.

    Parameters
    ----------
    on : bool
        Whether to enable deterministic mode.
    seed_value : int or None
        Random seed to use if deterministic mode is enabled.
    """
    global _deterministic, _random_seed
    _deterministic = on
    _random_seed = seed_value

    if _deterministic:
        import os
        os.environ["ITK_GLOBAL_DEFAULT_NUMBER_OF_THREADS"] = "1"

        import numpy as np
        import random

        if _random_seed is not None:
            np.random.seed(_random_seed)
            random.seed(_random_seed)