Console Output

+ python3 clubb_jax/tests/run_driver_extensions_test.py
CLUBB JAX runtime:
  Environment: /home/jenkins/workspace/clubb_branch_driver/.venv-jax (Python 3.12)
  Accelerator: cpu
  JAX: 0.11.0 (jaxlib 0.11.0)
  Device: cpu:0
An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.
Creating a custom hypergrid of parameters
 - Initial values file: '/home/jenkins/workspace/clubb_branch_driver/input/parameter_and_flag_configs/default/tunable_parameters.in'
 - Spec: C8/0.2:0.8/4
Writing to '/tmp/clubb-lifecycle-uf496t_p/bomex_multicol_params.in':
 - ngrdcol = 4
 - batch_size = 2
nzm = 88 -- nzt = 87
Initialized bomex case: nzm=88, nzt=87, ngrdcol=2
  dt_main=60.0s, time=0.0s to 480.0s, 8 steps
CLUBB cleanup complete.
JAX extensions: window equivalence and runtime batches pass