fix unittest

Author: chenyushuo

tests/explorer/workflow_test.py

@@ -731,12 +731,16 @@ async def mock_get_api_server_url_remote():
         async def mock_get_model_version_remote():
             return 1
 
+        async def mock_get_api_key_remote():
+            return "dummy_api_key"
+
         async def mock_get_model_config_remote():
             return InferenceModelConfig(model_path="dummy_model")
 
         model = MagicMock()
         model.get_api_server_url.remote = MagicMock(side_effect=mock_get_api_server_url_remote)
         model.get_model_version.remote = MagicMock(side_effect=mock_get_model_version_remote)
+        model.get_api_key.remote = MagicMock(side_effect=mock_get_api_key_remote)
         model.get_model_config.remote = MagicMock(side_effect=mock_get_model_config_remote)
 
         runner = WorkflowRunner(

trinity/trainer/verl/utils.py

@@ -100,7 +100,9 @@ def to_data_proto(
             )
     else:
         raise ValueError("Custom fields are not consistent across experiences.")
-    meta_info = {"model_versions": np.array([exp.info.get("model_version", 0) for exp in experiences])}
+    meta_info = {
+        "model_versions": np.array([exp.info.get("model_version", 0) for exp in experiences])
+    }
     return DataProto.from_single_dict(batch_dict, meta_info=meta_info)