Repository: ignite Updated Branches: refs/heads/master e142b100f -> 4bd2d8320
IGNITE-9788: ML: IgniteDataset op is not loaded in TensorFlow worker this closes #4912 Project: http://git-wip-us.apache.org/repos/asf/ignite/repo Commit: http://git-wip-us.apache.org/repos/asf/ignite/commit/4bd2d832 Tree: http://git-wip-us.apache.org/repos/asf/ignite/tree/4bd2d832 Diff: http://git-wip-us.apache.org/repos/asf/ignite/diff/4bd2d832 Branch: refs/heads/master Commit: 4bd2d8320911f1eaad800358056728bf5900dfc5 Parents: e142b10 Author: Anton Dmitriev <[email protected]> Authored: Thu Oct 4 17:03:51 2018 +0300 Committer: Yury Babak <[email protected]> Committed: Thu Oct 4 17:03:51 2018 +0300 ---------------------------------------------------------------------- .../cluster/tfrunning/TensorFlowServerScriptFormatter.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/ignite/blob/4bd2d832/modules/tensorflow/src/main/java/org/apache/ignite/tensorflow/cluster/tfrunning/TensorFlowServerScriptFormatter.java ---------------------------------------------------------------------- diff --git a/modules/tensorflow/src/main/java/org/apache/ignite/tensorflow/cluster/tfrunning/TensorFlowServerScriptFormatter.java b/modules/tensorflow/src/main/java/org/apache/ignite/tensorflow/cluster/tfrunning/TensorFlowServerScriptFormatter.java index 18854ab..ad77d16 100644 --- a/modules/tensorflow/src/main/java/org/apache/ignite/tensorflow/cluster/tfrunning/TensorFlowServerScriptFormatter.java +++ b/modules/tensorflow/src/main/java/org/apache/ignite/tensorflow/cluster/tfrunning/TensorFlowServerScriptFormatter.java @@ -66,7 +66,8 @@ public class TensorFlowServerScriptFormatter { builder.append("\n"); builder.append("import tensorflow as tf").append('\n'); - builder.append("fto_import_contrib_ops = tf.contrib.resampler").append("\n"); + builder.append("to_import_contrib_ops = tf.contrib.resampler").append("\n"); + builder.append("from tensorflow.contrib.ignite import IgniteDataset").append("\n"); builder.append("import tensorflow.contrib.igfs.python.ops.igfs_ops").append("\n"); builder.append("print('job:%s task:%d' % ('")
