/user/hominhduy.nguyen01/u24877/.local/lib/python3.10/site-packages/tyro/_parsers.py:379: UserWarning: The field `data.repo-id` is annotated with type `<class 'str'>`, but the default value `None` has type `<class 'NoneType'>`. We'll try to handle this gracefully, but it may cause unexpected behavior.
  warnings.warn(message)
/user/hominhduy.nguyen01/u24877/.local/lib/python3.10/site-packages/tyro/_parsers.py:379: UserWarning: The field `model.action-expert-variant` is annotated with type `typing.Literal['dummy', 'gemma_300m', 'gemma_2b', 'gemma_2b_lora']`, but the default value `gemma_300m_lora` has type `<class 'str'>`. We'll try to handle this gracefully, but it may cause unexpected behavior.
  warnings.warn(message)
04:26:49.173 [I] Running on: glogin9                                                              (167422:train.py:195)
Traceback (most recent call last):
  File "/mnt/vast-kisski/projects/kisski-spath/VLA_Groot/in_context_learning/ricl_openpi/scripts/train.py", line 273, in <module>
    main(_config.cli())
  File "/mnt/vast-kisski/projects/kisski-spath/VLA_Groot/in_context_learning/ricl_openpi/scripts/train.py", line 197, in main
    if config.batch_size % jax.device_count() != 0:
  File "/projects/extern/kisski/kisski-spath/dir.project/VLA_3D/miniconda3/envs/pitorch_doanh/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 1098, in device_count
    return int(get_backend(backend).device_count())
  File "/projects/extern/kisski/kisski-spath/dir.project/VLA_3D/miniconda3/envs/pitorch_doanh/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 1067, in get_backend
    return _get_backend_uncached(platform)
  File "/projects/extern/kisski/kisski-spath/dir.project/VLA_3D/miniconda3/envs/pitorch_doanh/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 1046, in _get_backend_uncached
    bs = backends()
  File "/projects/extern/kisski/kisski-spath/dir.project/VLA_3D/miniconda3/envs/pitorch_doanh/lib/python3.10/site-packages/jax/_src/xla_bridge.py", line 953, in backends
    assert _default_backend is not None
AssertionError