org.apache.ignite.tensorflow.cluster.util

Class TensorFlowUserScriptRunner

java.lang.Object

org.apache.ignite.tensorflow.core.util.AsyncNativeProcessRunner

org.apache.ignite.tensorflow.cluster.util.TensorFlowUserScriptRunner

public class TensorFlowUserScriptRunner
extends AsyncNativeProcessRunner

Utils class that helps to start and stop user script process.

Constructor Summary

Constructors

Constructor and Description
TensorFlowUserScriptRunner(Ignite ignite, ExecutorService executor, TensorFlowJobArchive jobArchive, TensorFlowClusterSpec clusterSpec, Consumer<String> out, Consumer<String> err) Constructs a new instance of TensorFlow user script runner.

Method Summary

Modifier and Type	Method and Description
void	doAfter() Method that should be called after starting the process.
NativeProcessRunner	doBefore() Method that should be called before starting the process.

Methods inherited from class org.apache.ignite.tensorflow.core.util.AsyncNativeProcessRunner

getException, isCompleted, start, stop

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

TensorFlowUserScriptRunner

public TensorFlowUserScriptRunner(Ignite ignite,
                                  ExecutorService executor,
                                  TensorFlowJobArchive jobArchive,
                                  TensorFlowClusterSpec clusterSpec,
                                  Consumer<String> out,
                                  Consumer<String> err)

Constructs a new instance of TensorFlow user script runner.

Parameters:

ignite - Ignite instance.

executor - Executor to be used in AsyncNativeProcessRunner.

jobArchive - Job archive that will be extracted and used as working directory for the native process.

clusterSpec - TensorFlow cluster specification.

out - Output stream data consumer.

err - Error stream data consumer.

Method Detail

doBefore

public NativeProcessRunner doBefore()

Method that should be called before starting the process.

Specified by:

doBefore in class AsyncNativeProcessRunner

Returns:

Prepared native process runner.

doAfter

public void doAfter()

Method that should be called after starting the process.

Specified by:

doAfter in class AsyncNativeProcessRunner