wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-29 03:04:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Java] Refine python function (#8943)

Browse Source
This commit is contained in:
chaokunyang
2020-06-16 16:22:49 +08:00
committed by GitHub
parent 14405b90d5
commit cb6f337372
13 changed files with 164 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files
streaming/java/streaming-runtime/src/main/java/io/ray/streaming/runtime/master/scheduler/controller/WorkerLifecycleController.java
+2 -2
View File
@@ -52,8 +52,8 @@ public class WorkerLifecycleController {
.setMaxRestarts(-1)
.remote();
} else {
actor = Ray.actor(new PyActorClass(
"ray.streaming.runtime.worker", "JobWorker"))
actor = Ray.actor(
PyActorClass.of("ray.streaming.runtime.worker", "JobWorker"))
.setResources(executionVertex.getResource())
.setMaxRestarts(-1)
.remote();
streaming/java/streaming-runtime/src/main/java/io/ray/streaming/runtime/rpc/RemoteCallWorker.java
+2 -2
View File
@@ -32,7 +32,7 @@ public class RemoteCallWorker {
// python
if (actor instanceof PyActorHandle) {
result = ((PyActorHandle) actor).task(new PyActorMethod("init", Object.class),
result = ((PyActorHandle) actor).task(PyActorMethod.of("init", Boolean.class),
context.getPythonWorkerContextBytes()).remote();
} else {
// java
@@ -56,7 +56,7 @@ public class RemoteCallWorker {
// python
if (actor instanceof PyActorHandle) {
result = ((PyActorHandle) actor)
.task(new PyActorMethod("start", Object.class)).remote();
.task(PyActorMethod.of("start", Boolean.class)).remote();
} else {
// java
result = ((ActorHandle<JobWorker>) actor).task(JobWorker::start).remote();