Repository: ignite Updated Branches: refs/heads/ignite-2.7 59060d375 -> 6af3af331
IGNITE-9788: ML: IgniteDataset op is not loaded in TensorFlow worker this closes #4912 (cherry picked from commit 4bd2d83) Project: http://git-wip-us.apache.org/repos/asf/ignite/repo Commit: http://git-wip-us.apache.org/repos/asf/ignite/commit/6af3af33 Tree: http://git-wip-us.apache.org/repos/asf/ignite/tree/6af3af33 Diff: http://git-wip-us.apache.org/repos/asf/ignite/diff/6af3af33 Branch: refs/heads/ignite-2.7 Commit: 6af3af33114b2bf1a5a03460f2dfe894f871c34f Parents: 59060d3 Author: Anton Dmitriev <[email protected]> Authored: Thu Oct 4 17:03:51 2018 +0300 Committer: Yury Babak <[email protected]> Committed: Thu Oct 4 17:04:33 2018 +0300 ---------------------------------------------------------------------- .../cluster/tfrunning/TensorFlowServerScriptFormatter.java | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/ignite/blob/6af3af33/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' % ('")
