diff --git a/src/ray/core_worker/test/core_worker_test.cc b/src/ray/core_worker/test/core_worker_test.cc index 4db8321722bb2..c62055bc45106 100644 --- a/src/ray/core_worker/test/core_worker_test.cc +++ b/src/ray/core_worker/test/core_worker_test.cc @@ -991,11 +991,11 @@ TEST(TestOverrideRuntimeEnv, TestOverrideEnvVars) { TEST(TestOverrideRuntimeEnv, TestPyModulesInherit) { rpc::RuntimeEnv child; auto parent = std::make_shared(); - parent->mutable_python_runtime_env()->mutable_dependent_modules()->Add("s3://456"); + parent->mutable_python_runtime_env()->mutable_py_modules()->Add("s3://456"); parent->mutable_uris()->mutable_py_modules_uris()->Add("s3://456"); auto result = CoreWorker::OverrideRuntimeEnv(child, parent); - ASSERT_EQ(result.python_runtime_env().dependent_modules().size(), 1); - ASSERT_EQ(result.python_runtime_env().dependent_modules()[0], "s3://456"); + ASSERT_EQ(result.python_runtime_env().py_modules().size(), 1); + ASSERT_EQ(result.python_runtime_env().py_modules()[0], "s3://456"); ASSERT_EQ(result.uris().py_modules_uris().size(), 1); ASSERT_EQ(result.uris().py_modules_uris()[0], "s3://456"); } @@ -1003,15 +1003,15 @@ TEST(TestOverrideRuntimeEnv, TestPyModulesInherit) { TEST(TestOverrideRuntimeEnv, TestOverridePyModules) { rpc::RuntimeEnv child; auto parent = std::make_shared(); - child.mutable_python_runtime_env()->mutable_dependent_modules()->Add("s3://123"); + child.mutable_python_runtime_env()->mutable_py_modules()->Add("s3://123"); child.mutable_uris()->mutable_py_modules_uris()->Add("s3://123"); - parent->mutable_python_runtime_env()->mutable_dependent_modules()->Add("s3://456"); - parent->mutable_python_runtime_env()->mutable_dependent_modules()->Add("s3://789"); + parent->mutable_python_runtime_env()->mutable_py_modules()->Add("s3://456"); + parent->mutable_python_runtime_env()->mutable_py_modules()->Add("s3://789"); parent->mutable_uris()->mutable_py_modules_uris()->Add("s3://456"); parent->mutable_uris()->mutable_py_modules_uris()->Add("s3://789"); auto result = CoreWorker::OverrideRuntimeEnv(child, parent); - ASSERT_EQ(result.python_runtime_env().dependent_modules().size(), 1); - ASSERT_EQ(result.python_runtime_env().dependent_modules()[0], "s3://123"); + ASSERT_EQ(result.python_runtime_env().py_modules().size(), 1); + ASSERT_EQ(result.python_runtime_env().py_modules()[0], "s3://123"); ASSERT_EQ(result.uris().py_modules_uris().size(), 1); ASSERT_EQ(result.uris().py_modules_uris()[0], "s3://123"); } diff --git a/src/ray/core_worker/test/mock_worker.cc b/src/ray/core_worker/test/mock_worker.cc index e8d4beca3e9a8..8365cd431afe7 100644 --- a/src/ray/core_worker/test/mock_worker.cc +++ b/src/ray/core_worker/test/mock_worker.cc @@ -106,7 +106,7 @@ class MockWorker { // Merge all the content from input args. std::vector buffer; for (const auto &arg : args) { - auto &data = arg->GetData(); + auto data = arg->GetData(); buffer.insert(buffer.end(), data->Data(), data->Data() + data->Size()); } if (buffer.size() >= 8) {