code: def test_get_mpi(device):
      """
      测试 jittor 多卡的初始化情况
      """
  
      model = JittorNormalModel_Classification_1(20, 10)
      with pytest.raises(NotImplementedError):
          driver = initialize_jittor_driver("jittor", device, model)
  
imports: import pytest
  from fastNLP.core.drivers import JittorSingleDriver, JittorMPIDriver
  from fastNLP.core.drivers.jittor_driver.initialize_jittor_driver import initialize_jittor_driver
  from tests.helpers.models.jittor_model import JittorNormalModel_Classification_1
  from fastNLP.envs.imports import _NEED_IMPORT_JITTOR
label_desc: Use image node_id 3 for calling a global function with example usage: test_get_mpi(device) without return types