Пример #1
0
def pytest_runtest_setup(item):
    if "JAX_ENABLE_X64" in os.environ:
        config.update("jax_enable_x64", True)
    set_rng_seed(0)
Пример #2
0
def pytest_runtest_setup(item):
    if 'JAX_ENABLE_x64' in os.environ:
        config.update('jax_enable_x64', True)
    set_rng_seed(0)
Пример #3
0
def pytest_runtest_setup(item):
    config.update('jax_platform_name', 'cpu')
    if 'JAX_ENABLE_x64' in os.environ:
        config.update('jax_enable_x64', True)
    set_rng_seed(0)
Пример #4
0
def pytest_runtest_setup(item):
    set_rng_seed(0)
Пример #5
0
def pytest_runtest_setup(item):
    config.update('jax_platform_name', 'cpu')
    set_rng_seed(0)